Glenn Jocher
GitHub
3 years ago
1 changed files with 1 additions and 1 deletions
+ 1
- 1
test.py
View File
| # Save JSON | # Save JSON | ||||
| if save_json and len(jdict): | if save_json and len(jdict): | ||||
| w = Path(weights[0] if isinstance(weights, list) else weights).stem if weights is not None else '' # weights | w = Path(weights[0] if isinstance(weights, list) else weights).stem if weights is not None else '' # weights | ||||
| anno_json = '../coco/annotations/instances_val2017.json' # annotations json | |||||
| anno_json = str(Path(data.get('path', '../coco')) / 'annotations/instances_val2017.json') # annotations json | |||||
| pred_json = str(save_dir / f"{w}_predictions.json") # predictions json | pred_json = str(save_dir / f"{w}_predictions.json") # predictions json | ||||
| print('\nEvaluating pycocotools mAP... saving %s...' % pred_json) | print('\nEvaluating pycocotools mAP... saving %s...' % pred_json) | ||||
| with open(pred_json, 'w') as f: | with open(pred_json, 'w') as f: |
Loading…