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

Made image size parameter more compatible with clean code 

Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens 2019-07-11 12:22:01 +02:00
parent 9fa525ebe3
commit 0ae33130ad
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -149,7 +149,7 @@ def predict(generator: callable,
ssd_model: tf.keras.models.Model,
use_dropout: bool,
forward_passes_per_image: int,
image_size: Tuple[int, int],
image_size: int,
output_path: str,
nr_digits: int) -> None:
"""
@ -175,6 +175,7 @@ def predict(generator: callable,
filename = f"dropout-{filename}"
output_file = os.path.join(output_path, filename)
label_output_file = os.path.join(output_path, label_filename)
image_size = (image_size, image_size)
batch_counter = 0
for x, filenames, inverse_transforms, original_labels in generator: