Fix default --weights yolov5s.pt (ultralytics#4458)

fhkiel-mlaip · Aug 17, 2021 · 996a0a5 · 996a0a5
1 parent 8ebe069
commit 996a0a5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/detect.py b/detect.py
@@ -244,7 +244,7 @@ def wrap_frozen_graph(gd, inputs, outputs):
 
 def parse_opt():
     parser = argparse.ArgumentParser()
-    parser.add_argument('--weights', nargs='+', type=str, default='yolov5s.pb', help='model.pt path(s)')
+    parser.add_argument('--weights', nargs='+', type=str, default='yolov5s.pt', help='model.pt path(s)')
     parser.add_argument('--source', type=str, default='data/images', help='file/dir/URL/glob, 0 for webcam')
     parser.add_argument('--imgsz', '--img', '--img-size', nargs='+', type=int, default=[640], help='inference size h,w')
     parser.add_argument('--conf-thres', type=float, default=0.25, help='confidence threshold')