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

Really fixed entropy threshold keyword argument

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-07-18 14:00:17 +02:00
parent a5555c5f73
commit 84df7b64fa
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -300,7 +300,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")
for entropy_threshold in entropy_thresholds: for entropy_threshold in entropy_thresholds:
decoded_predictions = decode_func(predictions, entropy_thresh=entropy_threshold) decoded_predictions = decode_func(predictions, entropy_threshold=entropy_threshold)
if not saved_images_decoding: if not saved_images_decoding:
custom_string = f"after-decoding-{entropy_threshold}" if use_entropy_threshold else "after-decoding" custom_string = f"after-decoding-{entropy_threshold}" if use_entropy_threshold else "after-decoding"
save_images(inputs, decoded_predictions, custom_string=custom_string) save_images(inputs, decoded_predictions, custom_string=custom_string)
@ -350,7 +350,7 @@ def _decode_predictions(predictions: np.ndarray,
img_width=image_size, img_width=image_size,
img_height=image_size, img_height=image_size,
input_coords="corners", input_coords="corners",
entropy_threshold=entropy_threshold entropy_thresh=entropy_threshold
) )