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

Added suffix to save function

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

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

@ -351,7 +351,8 @@ def _predict_loop(generator: Generator, use_dropout: bool, steps_per_epoch: int,
transformed_predictions = transform_func(decoded_predictions,
inverse_transforms)
save_func(transformed_predictions, original_labels, filenames,
batch_nr=batch_counter, entropy_threshold=entropy_threshold)
batch_nr=batch_counter, entropy_threshold=entropy_threshold,
suffix="_transformed")
if not saved_images_decoding:
saved_images_decoding = True
@ -486,10 +487,11 @@ def _transform_predictions(decoded_predictions: np.ndarray, inverse_transforms:
def _save_predictions(transformed_predictions: np.ndarray, original_labels: np.ndarray, filenames: Sequence[str],
output_file: str, label_output_file: str,
batch_nr: int, nr_digits: int, entropy_threshold: float) -> None:
batch_nr: int, nr_digits: int, entropy_threshold: float,
suffix: str) -> None:
counter_str = str(batch_nr).zfill(nr_digits)
filename = f"{output_file}-{counter_str}"
filename = f"{output_file}{suffix}-{counter_str}"
filename = f"{filename}-{entropy_threshold}" if entropy_threshold else filename
label_filename = f"{label_output_file}-{counter_str}.bin"