zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

export.py update

This commit is contained in:
Glenn Jocher 2020-07-01 16:15:25 -07:00
parent a62a1c2c67
commit 5323ad224d
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
detect.py
View File

@ -156,9 +156,9 @@ if __name__ == '__main__':
print(opt) print(opt)
with torch.no_grad(): with torch.no_grad():
# detect()
# Update all models
for opt.weights in ['yolov5s.pt', 'yolov5m.pt', 'yolov5l.pt', 'yolov5x.pt', 'yolov3-spp.pt']:
detect() 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)