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

Fixed indexing of instance array

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

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

@ -78,11 +78,10 @@ def save_ssd_train_images(images: np.ndarray, labels: np.ndarray,
ymax = instance[4] ymax = instance[4]
else: else:
instance = np.asarray(instance) instance = np.asarray(instance)
print(instance.shape)
class_id = np.argmax(instance[:-12], axis=0) class_id = np.argmax(instance[:-12], axis=0)
instance[-12:-8] *= instance[-4:] # multiply with variances instance[-12:-8] *= instance[-4:] # multiply with variances
instance[-11, -9] *= np.expand_dims(instance[-5] - instance[-7], axis=-1) instance[[-11, -9]] *= np.expand_dims(instance[-5] - instance[-7], axis=-1)
instance[-12, -10] *= np.expand_dims(instance[-6] - instance[-8], axis=-1) instance[[-12, -10]] *= np.expand_dims(instance[-6] - instance[-8], axis=-1)
instance[-12:-8] += instance[-8:-4] instance[-12:-8] += instance[-8:-4]
instance[-12:-8] *= image_size instance[-12:-8] *= image_size