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

Fixed saving debug images

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-07-09 14:21:38 +02:00
parent bdd0862d68
commit a0a976f9d7
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -55,10 +55,10 @@ def save_ssd_train_images(images: np.ndarray, labels: np.ndarray, output_path: s
current_axis = pyplot.gca() current_axis = pyplot.gca()
for instance in instances: for instance in instances:
xmin = instance[-12] * image_size xmin = (instance[-12] + instance[-8]) * image_size
ymin = instance[-11] * image_size ymin = (instance[-11] + instance[-7]) * image_size
xmax = instance[-10] * image_size xmax = (instance[-10] + instance[-6]) * image_size
ymax = instance[-9] * image_size ymax = (instance[-9] + instance[-5]) * image_size
class_id = np.argmax(instance[:-12], axis=0) class_id = np.argmax(instance[:-12], axis=0)
color = colors[class_id] color = colors[class_id]
label = f"{classes_to_names[class_id]}" label = f"{classes_to_names[class_id]}"