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

Save predictions directly after entropy threshold is applied as well

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-09-12 15:24:33 +02:00
parent c0e7b434dc
commit b861270e51
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -341,6 +341,9 @@ def _predict_loop(generator: Generator, use_dropout: bool, steps_per_epoch: int,
if use_dropout: if use_dropout:
decoded_predictions = apply_entropy_threshold_func(observations, decoded_predictions = apply_entropy_threshold_func(observations,
entropy_threshold=entropy_threshold) entropy_threshold=entropy_threshold)
save_func(decoded_predictions, original_labels, filenames,
batch_nr=batch_counter, entropy_threshold=entropy_threshold,
suffix="_entropy")
decoded_predictions = apply_top_k_func(decoded_predictions) decoded_predictions = apply_top_k_func(decoded_predictions)
else: else:
decoded_predictions = decode_func(predictions, entropy_threshold=entropy_threshold) decoded_predictions = decode_func(predictions, entropy_threshold=entropy_threshold)