Add RF-DETR-seg model (#154)

Add RF-DETR models:
- rf-detr-nano
- rf-detr-small
- rf-detr-medium
- rf-detr-seg-preview

Author: jamjamjon

examples/rfdetr/README.md

@@ -1,17 +1,6 @@
 ## Quick Start
 
 ```shell
-cargo run -r --example rfdetr
-```
-
-## Results
-
-```
-[Bboxes]: Found 5 objects
-0: Bbox { xyxy: [47.969677, 397.81808, 246.22426, 904.8823], class_id: 0, name: Some("person"), confidence: 0.94432133 }
-1: Bbox { xyxy: [668.0796, 399.28854, 810.3779, 880.7412], class_id: 0, name: Some("person"), confidence: 0.93386495 }
-2: Bbox { xyxy: [20.852705, 229.30482, 807.43494, 729.51196], class_id: 5, name: Some("bus"), confidence: 0.9319465 }
-3: Bbox { xyxy: [223.28226, 405.37265, 343.92603, 859.50366], class_id: 0, name: Some("person"), confidence: 0.9130827 }
-4: Bbox { xyxy: [0.0, 552.6165, 65.99908, 868.00525], class_id: 0, name: Some("person"), confidence: 0.7910869 }
-
+cargo run --example rfdetr -- --scale n --dtype f16     # detection
+cargo run --example rfdetr -- --dtype f16 --task seg    # instance segment
 ```

examples/rfdetr/main.rs

@@ -1,14 +1,50 @@
 use anyhow::Result;
-use usls::{models::RFDETR, Annotator, Config, DataLoader};
+use usls::{models::RFDETR, Annotator, Config, DataLoader, Scale, Task};
+
+#[derive(argh::FromArgs)]
+/// Example
+struct Args {
+    /// device
+    #[argh(option, default = "String::from(\"cpu\")")]
+    device: String,
+
+    /// task
+    #[argh(option, default = "String::from(\"det\")")]
+    task: String,
+
+    /// scale
+    #[argh(option, default = "String::from(\"n\")")]
+    scale: String,
+
+    /// dtype
+    #[argh(option, default = "String::from(\"auto\")")]
+    dtype: String,
+}
 
 fn main() -> Result<()> {
     tracing_subscriber::fmt()
         .with_env_filter(tracing_subscriber::EnvFilter::from_default_env())
         .with_timer(tracing_subscriber::fmt::time::ChronoLocal::rfc_3339())
         .init();
+    let args: Args = argh::from_env();
 
     // config
-    let mut model = RFDETR::new(Config::rfdetr_base().commit()?)?;
+    let config = match args.task.parse()? {
+        Task::ObjectDetection => match args.scale.parse()? {
+            Scale::N => Config::rfdetr_nano(),
+            Scale::S => Config::rfdetr_small(),
+            Scale::M => Config::rfdetr_medium(),
+            Scale::B => Config::rfdetr_base(),
+            Scale::L => Config::rfdetr_large(),
+            _ => unimplemented!("Unsupported model scale: {:?}. Try b, s, t.", args.scale),
+        },
+        Task::InstanceSegmentation => Config::rfdetr_seg_preview(),
+        _ => unimplemented!("Unsupported task: {:?}. Try det, seg.", args.task),
+    }
+    .with_dtype_all(args.dtype.parse()?)
+    .with_device_all(args.device.parse()?)
+    .commit()?;
+    let mut model = RFDETR::new(config)?;
 
     // load
     let xs = DataLoader::try_read_n(&["./assets/bus.jpg"])?;