@ -78,7 +78,7 @@ 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)
|
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)
|
||||||
|
|||||||
Reference in New Issue
Block a user