Glenn Jocher
4 years ago
1 changed files with 4 additions and 3 deletions
+ 4
- 3
test.py
View File
| save_json=False, | save_json=False, | ||||
| single_cls=False, | single_cls=False, | ||||
| augment=False, | augment=False, | ||||
| verbose=False, | |||||
| model=None, | model=None, | ||||
| dataloader=None, | dataloader=None, | ||||
| fast=False, | |||||
| verbose=False): | |||||
| fast=False): | |||||
| # Initialize/load model and set device | # Initialize/load model and set device | ||||
| if model is None: | if model is None: | ||||
| training = False | training = False | ||||
| opt.iou_thres, | opt.iou_thres, | ||||
| opt.save_json, | opt.save_json, | ||||
| opt.single_cls, | opt.single_cls, | ||||
| opt.augment) | |||||
| opt.augment, | |||||
| opt.verbose) | |||||
| elif opt.task == 'study': # run over a range of settings and save/plot | elif opt.task == 'study': # run over a range of settings and save/plot | ||||
| for weights in ['yolov5s.pt', 'yolov5m.pt', 'yolov5l.pt', 'yolov5x.pt']: | for weights in ['yolov5s.pt', 'yolov5m.pt', 'yolov5l.pt', 'yolov5x.pt']: |
Loading…