export.py update

5 years ago · 5323ad224d
parent a62a1c2c67
commit 5323ad224d
1 changed files with 5 additions and 5 deletions
--- a/detect.py
+++ b/detect.py
@ -156,9 +156,9 @@ if __name__ == '__main__':
    print(opt)
    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)