Merge pull request #1291 from mikel-brostrom/pass-conf-n-iou-to-yolonas

pass conf and iou to yolonas
mikel-brostrom · Feb 7, 2024 · 09c4f22 · 09c4f22
2 parents 7157d28 + ad35b38
commit 09c4f22
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/detectors/yolonas.py b/examples/detectors/yolonas.py
@@ -65,8 +65,8 @@ def __call__(self, im, augment, visualize):
         with torch.no_grad():
             preds = self.model.predict(
                 im,
-                iou=0.5,
-                conf=0.7,
+                iou=self.args.iou,
+                conf=self.args.conf,
                 fuse_model=False
             )[0].prediction