Updated debug print

Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
2019-07-10 17:33:15 +02:00
parent 82689ba6ec
commit b5058d9ee9

View File

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