Ankle/yolov5_birdproject
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-08-17 23:29:07 +02:00
1 parent 808bcad3bb
commit 6dd7dd8dd3
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
detect.py
+1 -1
View File
@@ -244,7 +244,7 @@ def run(weights='yolov5s.pt', # model.pt path(s)
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')