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

Prevented overwriting of saved images

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-07-24 12:04:44 +02:00
parent 5cba4046a5
commit 761f6124e0
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/twomartens/masterthesis/ssd.py
View File

@ -27,9 +27,11 @@ Functions:
import functools import functools
import os import os
import pickle import pickle
from typing import Generator
from typing import List, Sequence, Tuple, Generator from typing import List
from typing import Optional from typing import Optional
from typing import Sequence
from typing import Tuple
import math import math
import numpy as np 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: if not saved_images_prediction:
save_images(inputs, predictions, custom_string="after-prediction") save_images(inputs, predictions, custom_string="after-prediction")
saved_images_prediction = True
for entropy_threshold in entropy_thresholds: for entropy_threshold in entropy_thresholds:
decoded_predictions = decode_func(predictions, entropy_threshold=entropy_threshold) decoded_predictions = decode_func(predictions, entropy_threshold=entropy_threshold)
if not saved_images_decoding: if not saved_images_decoding: