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

Changed code to save images with predictions not ground truth 

Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens 2019-07-11 17:13:17 +02:00
parent 6f6b4b7e34
commit c2be08e0c8
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

@ -325,7 +325,7 @@ def _ssd_evaluate(args: argparse.Namespace) -> None:
predictions = _ssd_evaluate_unbatch_list(predictions_glob_string)
_pickle(predictions_file, predictions)
_ssd_evaluate_save_images(filenames, labels,
_ssd_evaluate_save_images(filenames, predictions,
coco_utils.get_coco_category_maps, debug.save_ssd_train_images,
image_size, output_path, coco_path)