This commit is contained in:
Glenn Jocher 2019-04-02 14:19:53 +02:00
parent af61da5d41
commit 330caefe69
1 changed files with 2 additions and 2 deletions

View File

@ -40,8 +40,8 @@ def test(
# Configure run
data_cfg = parse_data_cfg(data_cfg)
test_path = data_cfg['valid']
if (os.sep + 'coco' + os.sep) in test_path: # COCO dataset probable
save_json = True # use pycocotools
# if (os.sep + 'coco' + os.sep) in test_path: # COCO dataset probable
# save_json = True # use pycocotools
# Dataloader
dataset = LoadImagesAndLabels(test_path, img_size=img_size)