Fix default `--weights yolov5s.pt` (#4458)
Browse files
detect.py
CHANGED
@@ -244,7 +244,7 @@ def run(weights='yolov5s.pt', # model.pt path(s)
|
|
244 |
|
245 |
def parse_opt():
|
246 |
parser = argparse.ArgumentParser()
|
247 |
-
parser.add_argument('--weights', nargs='+', type=str, default='yolov5s.
|
248 |
parser.add_argument('--source', type=str, default='data/images', help='file/dir/URL/glob, 0 for webcam')
|
249 |
parser.add_argument('--imgsz', '--img', '--img-size', nargs='+', type=int, default=[640], help='inference size h,w')
|
250 |
parser.add_argument('--conf-thres', type=float, default=0.25, help='confidence threshold')
|
|
|
244 |
|
245 |
def parse_opt():
|
246 |
parser = argparse.ArgumentParser()
|
247 |
+
parser.add_argument('--weights', nargs='+', type=str, default='yolov5s.pt', help='model.pt path(s)')
|
248 |
parser.add_argument('--source', type=str, default='data/images', help='file/dir/URL/glob, 0 for webcam')
|
249 |
parser.add_argument('--imgsz', '--img', '--img-size', nargs='+', type=int, default=[640], help='inference size h,w')
|
250 |
parser.add_argument('--conf-thres', type=float, default=0.25, help='confidence threshold')
|