🐍 v0.1.0 (#53)

examples/fast/README.md

@@ -0,0 +1,6 @@
+## Quick Start
+
+```shell
+cargo run -r --example fast
+```
+

examples/fast/main.rs

@@ -0,0 +1,65 @@
+use anyhow::Result;
+use usls::{models::DB, Annotator, DataLoader, Options, Scale};
+
+#[derive(argh::FromArgs)]
+/// Example
+struct Args {
+    /// device
+    #[argh(option, default = "String::from(\"cpu:0\")")]
+    device: String,
+
+    /// scale
+    #[argh(option, default = "String::from(\"t\")")]
+    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();
+
+    // build model
+    let options = match args.scale.as_str().try_into()? {
+        Scale::T => Options::fast_tiny(),
+        Scale::S => Options::fast_small(),
+        Scale::B => Options::fast_base(),
+        _ => unimplemented!("Unsupported model scale: {:?}. Try b, s, t.", args.scale),
+    };
+    let mut model = DB::new(
+        options
+            .with_model_dtype(args.dtype.as_str().try_into()?)
+            .with_model_device(args.device.as_str().try_into()?)
+            .commit()?,
+    )?;
+
+    // load image
+    let x = DataLoader::try_read_batch(&[
+        "images/table.png",
+        "images/table1.jpg",
+        "images/table2.png",
+        "images/table-ch.jpg",
+        "images/db.png",
+        "images/street.jpg",
+    ])?;
+
+    // run
+    let y = model.forward(&x)?;
+
+    // annotate
+    let annotator = Annotator::default()
+        .without_bboxes(true)
+        .without_mbrs(true)
+        .with_polygons_alpha(60)
+        .with_contours_color([255, 105, 180, 255])
+        .with_saveout(model.spec());
+    annotator.annotate(&x, &y);
+
+    Ok(())
+}