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

Only the first batch size images are saved on evaluate 

Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens 2019-08-11 15:36:39 +02:00
parent 6e5cbd3eb3
commit 588e66c772
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/twomartens/masterthesis/cli.py
View File

@ -386,7 +386,7 @@ def _ssd_evaluate_save_images(filenames: Sequence[str], labels: Sequence[np.ndar
image_size: int, batch_size: int,
output_path: str, coco_path: str) -> None:
save_images(filenames, labels, output_path, coco_path, image_size, get_coco_cat_maps_func)
save_images(filenames[:batch_size], labels[:batch_size], output_path, coco_path, image_size, get_coco_cat_maps_func)
def _visualise_gt(args: argparse.Namespace,