Use one trajectory only for prediction
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
@ -341,7 +341,8 @@ def _ssd_test(args: argparse.Namespace) -> None:
|
||||
predictor_sizes=ssd_model.predictor_sizes,
|
||||
batch_size=batch_size,
|
||||
resized_shape=(image_size, image_size),
|
||||
training=False, evaluation=True, augment=False)
|
||||
training=False, evaluation=True, augment=False,
|
||||
nr_trajectories=1)
|
||||
del file_names_photos, instances
|
||||
|
||||
nr_digits = math.ceil(math.log10(math.ceil(length_dataset / batch_size)))
|
||||
|
||||
Reference in New Issue
Block a user