From 761f6124e02a8c96b5d049b5b464bbfd1663d43e Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Wed, 24 Jul 2019 12:04:44 +0200 Subject: [PATCH] Prevented overwriting of saved images Signed-off-by: Jim Martens --- src/twomartens/masterthesis/ssd.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/src/twomartens/masterthesis/ssd.py b/src/twomartens/masterthesis/ssd.py index 6a8d034..08a3ea2 100644 --- a/src/twomartens/masterthesis/ssd.py +++ b/src/twomartens/masterthesis/ssd.py @@ -27,9 +27,11 @@ Functions: import functools import os import pickle - -from typing import List, Sequence, Tuple, Generator +from typing import Generator +from typing import List from typing import Optional +from typing import Sequence +from typing import Tuple import math import numpy as np @@ -299,6 +301,7 @@ def _predict_loop(generator: Generator, use_dropout: bool, steps_per_epoch: int, if not saved_images_prediction: save_images(inputs, predictions, custom_string="after-prediction") + saved_images_prediction = True for entropy_threshold in entropy_thresholds: decoded_predictions = decode_func(predictions, entropy_threshold=entropy_threshold) if not saved_images_decoding: