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 # Configure run
data_cfg = parse_data_cfg(data_cfg) data_cfg = parse_data_cfg(data_cfg)
test_path = data_cfg['valid'] test_path = data_cfg['valid']
if (os.sep + 'coco' + os.sep) in test_path: # COCO dataset probable # if (os.sep + 'coco' + os.sep) in test_path: # COCO dataset probable
save_json = True # use pycocotools # save_json = True # use pycocotools
# Dataloader # Dataloader
dataset = LoadImagesAndLabels(test_path, img_size=img_size) dataset = LoadImagesAndLabels(test_path, img_size=img_size)