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