feat: Remove bbox crate and use 1024 for image size

2025-08-05 18:14:31 +05:30
parent 06fb0b4487
commit 043a845fc1
10 changed files with 89 additions and 839 deletions
--- a/src/cli.rs
+++ b/src/cli.rs
@@ -18,6 +18,7 @@ pub enum SubCommand {
 #[derive(Debug, clap::ValueEnum, Clone, Copy)]
 pub enum Models {
    RetinaFace,
+    Yolo,
 }

 #[derive(Debug, clap::ValueEnum, Clone, Copy)]
@@ -49,6 +50,8 @@ pub struct Detect {
    pub output: Option<PathBuf>,
    #[clap(short, long, default_value_t = 0.8)]
    pub threshold: f32,
+    #[clap(short, long, default_value_t = 0.3)]
+    pub nms_threshold: f32,
    pub image: PathBuf,
 }