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

Only save images from first batch in evaluate

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-07-11 17:19:56 +02:00
parent c2be08e0c8
commit 38229e1848
1 changed files with 13 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -327,7 +327,8 @@ def _ssd_evaluate(args: argparse.Namespace) -> None:
_ssd_evaluate_save_images(filenames, predictions, _ssd_evaluate_save_images(filenames, predictions,
coco_utils.get_coco_category_maps, debug.save_ssd_train_images, coco_utils.get_coco_category_maps, debug.save_ssd_train_images,
image_size, output_path, coco_path) image_size, batch_size,
output_path, coco_path)
predictions_per_class = evaluate.prepare_predictions(predictions, nr_classes) predictions_per_class = evaluate.prepare_predictions(predictions, nr_classes)
_pickle(predictions_per_class_file, predictions_per_class) _pickle(predictions_per_class_file, predictions_per_class)
@ -365,16 +366,24 @@ def _ssd_evaluate(args: argparse.Namespace) -> None:
def _ssd_evaluate_save_images(filenames: Sequence[str], labels: Sequence[np.ndarray], def _ssd_evaluate_save_images(filenames: Sequence[str], labels: Sequence[np.ndarray],
get_coco_cat_maps_func: callable, save_images: callable, get_coco_cat_maps_func: callable, save_images: callable,
image_size: int, image_size: int, batch_size: int,
output_path: str, coco_path: str) -> None: output_path: str, coco_path: str) -> None:
from PIL import Image from PIL import Image
images = [] images = []
for filename in filenames: for i, filename in enumerate(filenames):
if i == batch_size:
break
with Image.open(filename) as image: with Image.open(filename) as image:
images.append(np.array(image, dtype=np.uint8)) images.append(np.array(image, dtype=np.uint8))
save_images(images, labels, output_path, coco_path, image_size, get_coco_cat_maps_func) cleaned_labels = []
for j, label in enumerate(labels):
if j == batch_size:
break
cleaned_labels.append(label)
save_images(images, cleaned_labels, output_path, coco_path, image_size, get_coco_cat_maps_func)
def _init_eager_mode() -> None: def _init_eager_mode() -> None: