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

Properly reversed encoding of labels to visualize them

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

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

@ -78,10 +78,16 @@ 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]) * image_size instance[-12:-8] *= instance[-4:] # multiply with variances
ymin = (instance[-11] + instance[-7]) * image_size instance[-11, -9] *= np.expand_dims(instance[-5] - instance[-7], axis=-1)
xmax = (instance[-10] + instance[-6]) * image_size instance[-12, -10] *= np.expand_dims(instance[-6] - instance[-8], axis=-1)
ymax = (instance[-9] + instance[-5]) * image_size instance[-12:-8] += instance[-8:-4]
instance[-12:-8] *= image_size
xmin = instance[-12]
ymin = instance[-11]
xmax = instance[-10]
ymax = instance[-9]
if class_id == 0: if class_id == 0:
continue continue