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

Fixed calculation of bbox coordinates

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

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

@ -78,19 +78,14 @@ def save_ssd_train_images(images: np.ndarray, labels: np.ndarray,
ymax = instance[4] ymax = instance[4]
else: else:
class_id = np.argmax(instance[:-12], axis=0) class_id = np.argmax(instance[:-12], axis=0)
xmin = instance[-12] + instance[-8] xmin = (instance[-12] + instance[-8]) * image_size
ymin = instance[-11] + instance[-7] ymin = (instance[-11] + instance[-7]) * image_size
xmax = instance[-10] + instance[-6] xmax = (instance[-10] + instance[-6]) * image_size
ymax = instance[-9] + instance[-5] ymax = (instance[-9] + instance[-5]) * image_size
if class_id == 0: if class_id == 0:
continue continue
xmin *= image_size
ymin *= image_size
xmax *= image_size
ymax *= image_size
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(