2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Added conversion to int for class_id

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-07-10 16:01:18 +02:00
parent aaec64786c
commit 67c098b0d6
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/twomartens/masterthesis/debug.py
View File

@ -71,7 +71,7 @@ def save_ssd_train_images(images: np.ndarray, labels: np.ndarray,
for instance in instances: for instance in instances:
if len(instance) == 5: if len(instance) == 5:
class_id = instance[0] class_id = int(instance[0])
xmin = instance[1] xmin = instance[1]
ymin = instance[2] ymin = instance[2]
xmax = instance[3] xmax = instance[3]
@ -91,7 +91,6 @@ def save_ssd_train_images(images: np.ndarray, labels: np.ndarray,
xmax *= image_size xmax *= image_size
ymax *= image_size ymax *= image_size
print(class_id)
color = colors[class_id] color = colors[class_id]
label = f"{classes_to_names[class_id]}" label = f"{classes_to_names[class_id]}"
current_axis.add_patch( current_axis.add_patch(