This commit is contained in:
Glenn Jocher committed 2020-06-24 13:20:14 -07:00
1 parent e071b8dd3a
commit 227aa735dc
2 files changed
+6 -1

No files matched your search

+5
View File
@@ -156,3 +156,8 @@ if __name__ == '__main__':
with torch.no_grad():
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)