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

export.py update

Browse Source
This commit is contained in:
Glenn Jocher committed 2020-07-01 16:14:49 -07:00
1 parent f1d67f4110
commit a62a1c2c67
2 files changed
+9 -9
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

detect.py
+4 -4
View File
@@ -156,9 +156,9 @@ if __name__ == '__main__':
print(opt)
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)
for opt.weights in ['yolov5s.pt', 'yolov5m.pt', 'yolov5l.pt', 'yolov5x.pt', 'yolov3-spp.pt']:
detect()
create_pretrained(opt.weights, opt.weights)