Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
test.py
View File

@ -188,7 +188,7 @@ def test(cfg,
# Save JSON # Save JSON
if save_json and map and len(jdict): 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] imgIds = [int(Path(x).stem.split('_')[-1]) for x in dataloader.dataset.img_files]
with open('results.json', 'w') as file: with open('results.json', 'w') as file:
json.dump(jdict, file) json.dump(jdict, file)