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

Changed pre-trained weights file to sub-sampled one

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-08-11 13:49:46 +02:00
parent 5a0ae208d3
commit 72c821dc2a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -702,7 +702,7 @@ def _ssd_test_prepare_paths(args: argparse.Namespace,
output_path = f"{output_path}/{args.network}/test/{args.iteration}/"
checkpoint_path = f"{weights_path}/{args.network}/train/{args.train_iteration}"
if test_pretrained:
weights_file = f"{weights_path}/{args.network}/VGG_coco_SSD_300x300_iter_400000.h5"
weights_file = f"{weights_path}/{args.network}/VGG_coco_SSD_300x300_iter_400000_subsampled.h5"
else:
weights_file = f"{checkpoint_path}/ssd300_weights.h5"