Change thres and thresh arguments to threshold

tryolabs · Aug 22, 2022 · b2b8d0d · b2b8d0d
1 parent 1ef4d5f
commit b2b8d0d
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/demos/yolov7/README.md b/demos/yolov7/README.md
@@ -13,7 +13,7 @@ Simplest possible example of tracking. Based on [Yolov7](https://github.com/Wong
     ```bash
         python demo.py --help 
     ``` 
-   Bonus: Use additional arguments `--detector-path`, `--img-size`, `--iou-thres`,`--conf-thres`, `--classes`, `--track-points` as you wish.
+   Bonus: Use additional arguments `--detector-path`, `--img-size`, `--iou-threshold`,`--conf-threshold`, `--classes`, `--track-points` as you wish.
 
 
 ## Explanation

diff --git a/demos/yolov7/src/demo.py b/demos/yolov7/src/demo.py
@@ -146,8 +146,8 @@ def yolo_detections_to_norfair_detections(
 parser.add_argument("files", type=str, nargs="+", help="Video files to process")
 parser.add_argument("--detector-path", type=str, default="/yolov7.pt", help="YOLOv7 model path")
 parser.add_argument("--img-size", type=int, default="720", help="YOLOv7 inference size (pixels)")
-parser.add_argument("--conf-thres", type=float, default="0.25", help="YOLOv7 object confidence threshold")
-parser.add_argument("--iou-thresh", type=float, default="0.45", help="YOLOv7 IOU threshold for NMS")
+parser.add_argument("--conf-threshold", type=float, default="0.25", help="YOLOv7 object confidence threshold")
+parser.add_argument("--iou-threshold", type=float, default="0.45", help="YOLOv7 IOU threshold for NMS")
 parser.add_argument("--classes", nargs="+", type=int, help="Filter by class: --classes 0, or --classes 0 2 3")
 parser.add_argument("--device", type=str, default=None, help="Inference device: 'cpu' or 'cuda'")
 parser.add_argument("--track-points", type=str, default="centroid", help="Track points: 'centroid' or 'bbox'")
@@ -174,8 +174,8 @@ def yolo_detections_to_norfair_detections(
     for frame in video:
         yolo_detections = model(
             frame,
-            conf_threshold=args.conf_thres,
-            iou_threshold=args.iou_thresh,
+            conf_threshold=args.conf_threshold,
+            iou_threshold=args.iou_threshold,
             image_size=args.img_size,
             classes=args.classes
         )