Glenn Jocher
4 years ago
1 changed files with 5 additions and 5 deletions
+ 5
- 5
detect.py
View File
| print(opt) | print(opt) | ||||
| with torch.no_grad(): | with torch.no_grad(): | ||||
| # detect() | |||||
| detect() | |||||
| # Update all models | |||||
| for opt.weights in ['yolov5s.pt', 'yolov5m.pt', 'yolov5l.pt', 'yolov5x.pt', 'yolov3-spp.pt']: | |||||
| detect() | |||||
| create_pretrained(opt.weights, opt.weights) | |||||
| # # Update all models | |||||
| # for opt.weights in ['yolov5s.pt', 'yolov5m.pt', 'yolov5l.pt', 'yolov5x.pt', 'yolov3-spp.pt']: | |||||
| # detect() | |||||
| # create_pretrained(opt.weights, opt.weights) |
Loading…