From 325e12faa18bc948e39b838bf856418dcb4b3bd8 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 17 Aug 2021 23:28:31 +0200 Subject: [PATCH] Fix default `--weights yolov5s.pt` --- detect.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/detect.py b/detect.py index a2331e23b43e..601d5daf9852 100644 --- 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')