Ankle/yolov5_birdproject
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

export.py update

Browse Source
This commit is contained in:
Glenn Jocher committed 2020-07-01 16:15:25 -07:00
1 parent a62a1c2c67
commit 5323ad224d
1 file changed
+5 -5
Download Patch File Download Diff File Expand all files Collapse all files
detect.py
+5 -5
View File
@@ -156,9 +156,9 @@ if __name__ == '__main__':
print(opt) print(opt)
with torch.no_grad(): with torch.no_grad():
# detect() detect()
# Update all models # # Update all models
for opt.weights in ['yolov5s.pt', 'yolov5m.pt', 'yolov5l.pt', 'yolov5x.pt', 'yolov3-spp.pt']: # for opt.weights in ['yolov5s.pt', 'yolov5m.pt', 'yolov5l.pt', 'yolov5x.pt', 'yolov3-spp.pt']:
detect() # detect()
create_pretrained(opt.weights, opt.weights) # create_pretrained(opt.weights, opt.weights)