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

Fixed calculation of number of digits

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-04-29 11:56:08 +02:00
parent 83bd83e3d1
commit a7a269ab73
2 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/twomartens/masterthesis/data.py
View File

@ -282,7 +282,7 @@ def load_scenenet_val(photo_paths: Sequence[Sequence[str]],
dataset = dataset.batch(batch_size=batch_size) dataset = dataset.batch(batch_size=batch_size)
dataset = dataset.map(_load_images_ssd_callback(resized_shape)) dataset = dataset.map(_load_images_ssd_callback(resized_shape))
nr_digits = math.ceil((length_dataset * num_epochs) / batch_size) nr_digits = math.ceil(math.log10(math.ceil((length_dataset * num_epochs) / batch_size)))
return dataset, nr_digits return dataset, nr_digits

1
src/twomartens/masterthesis/main.py
View File

@ -190,7 +190,6 @@ def _ssd_val(args: argparse.Namespace) -> None:
scenenet_data, nr_digits = data.load_scenenet_val(file_names_photos, instances, args.coco_path, scenenet_data, nr_digits = data.load_scenenet_val(file_names_photos, instances, args.coco_path,
batch_size=batch_size, resized_shape=(image_size, image_size)) batch_size=batch_size, resized_shape=(image_size, image_size))
print(nr_digits)
use_summary_writer = summary_ops_v2.create_file_writer( use_summary_writer = summary_ops_v2.create_file_writer(
f"{args.summary_path}/val/ssd/{args.iteration}" f"{args.summary_path}/val/ssd/{args.iteration}"
) )