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

Update utils.py strip_optimizer() (#509) 

Follow-on update that I missed adding into PR 509.
This commit is contained in:
Glenn Jocher 2020-07-24 19:30:37 -07:00 committed by GitHub
parent 0032af2980
commit fe094333e8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
detect.py
View File

@ -156,6 +156,6 @@ if __name__ == '__main__':
if opt.update: # update all models (to fix SourceChangeWarning)
for opt.weights in ['yolov5s.pt', 'yolov5m.pt', 'yolov5l.pt', 'yolov5x.pt', 'yolov3-spp.pt']:
detect()
create_pretrained(opt.weights, opt.weights)
strip_optimizer(opt.weights)
else:
detect()