This commit is contained in:
Glenn Jocher 2020-03-08 12:35:04 -07:00
parent 952df070db
commit 23389da9ec
1 changed files with 1 additions and 1 deletions

View File

@ -188,7 +188,7 @@ def test(cfg,
# Save JSON
if save_json and map and len(jdict):
print('COCO mAP with pycocotools...')
print('\nCOCO mAP with pycocotools...')
imgIds = [int(Path(x).stem.split('_')[-1]) for x in dataloader.dataset.img_files]
with open('results.json', 'w') as file:
json.dump(jdict, file)