updates
This commit is contained in:
parent
af61da5d41
commit
330caefe69
4
test.py
4
test.py
|
@ -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)
|
||||
|
|
Loading…
Reference in New Issue