Spaces:

xiang-wuu
/

yolov5

Runtime error

glenn-jocher commited on Jul 1, 2020

Commit

5323ad2

•

1 Parent(s): a62a1c2

export.py update

Files changed (1) hide show

detect.py CHANGED Viewed

@@ -156,9 +156,9 @@ if __name__ == '__main__':
     print(opt)
     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)

     print(opt)
     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)