diff --git a/pyro-predictor/pyro_predictor/vision.py b/pyro-predictor/pyro_predictor/vision.py index 0e30bcb4..e90b3a24 100644 --- a/pyro-predictor/pyro_predictor/vision.py +++ b/pyro-predictor/pyro_predictor/vision.py @@ -20,9 +20,9 @@ __all__ = ["Classifier"] -MODEL_REPO_ID = "pyronear/yolo11s_plucky-pelican_v7.1.0" +MODEL_REPO_ID = "pyronear/yolo11s_quick-quokka_v8.0.0" MODEL_NAME = "ncnn_cpu.tar.gz" -MODEL_SLUG = MODEL_REPO_ID.split("/", 1)[1] # e.g. "yolo11s_plucky-pelican_v7.1.0" +MODEL_SLUG = MODEL_REPO_ID.split("/", 1)[1] MODEL_CACHE_SUBDIR = "models" logging.basicConfig(format="%(asctime)s | %(levelname)s: %(message)s", level=logging.INFO, force=True) diff --git a/src/run.py b/src/run.py index 982613bc..55cabdf4 100644 --- a/src/run.py +++ b/src/run.py @@ -95,7 +95,7 @@ def main(args): ) # Model parser.add_argument("--model_path", type=str, default=None, help="model path") - parser.add_argument("--thresh", type=float, default=0.35, help="Confidence threshold") + parser.add_argument("--thresh", type=float, default=0.25, help="Confidence threshold") parser.add_argument("--max_bbox_size", type=float, default=0.4, help="Maximum bbox size") # Camera & cache @@ -119,7 +119,7 @@ def main(args): parser.add_argument( "--nb-consecutive_frames", type=int, - default=5, + default=6, help="Number of consecutive frames to combine for prediction", ) parser.add_argument(