🐍 v0.1.0 (#53)

examples/mobileone/README.md

@@ -0,0 +1,13 @@
+## Quick Start
+
+```shell
+cargo run -r -F cuda --example mobileone -- --device cuda --dtype fp16
+```
+ 
+
+```shell
+0: Y { Probs:  { Top5: [(263, 0.6109131, Some("Pembroke, Pembroke Welsh corgi")), (264, 0.2062352, Some("Cardigan, Cardigan Welsh corgi")), (231, 0.028572788, Some("collie")), (273, 0.015174894, Some("dingo, warrigal, warragal, Canis dingo")), (248, 0.014367299, Some("Eskimo dog, husky"))] } }
+1: Y { Probs:  { Top5: [(284, 0.9907692, Some("siamese cat, Siamese")), (285, 0.0015794479, Some("Egyptian cat")), (174, 0.0015189401, Some("Norwegian elkhound, elkhound")), (225, 0.00031838714, Some("malinois")), (17, 0.00027021166, Some("jay"))] } }
+2: Y { Probs:  { Top5: [(387, 0.94238573, Some("lesser panda, red panda, panda, bear cat, cat bear, Ailurus fulgens")), (368, 0.0029994072, Some("gibbon, Hylobates lar")), (277, 0.0016564301, Some("red fox, Vulpes vulpes")), (356, 0.0015081967, Some("weasel")), (295, 0.001427932, Some("American black bear, black bear, Ursus americanus, Euarctos americanus"))] } }
+
+```

examples/mobileone/main.rs

@@ -0,0 +1,57 @@
+use usls::{models::ImageClassifier, Annotator, DataLoader, Options};
+
+#[derive(argh::FromArgs)]
+/// Example
+struct Args {
+    /// dtype
+    #[argh(option, default = "String::from(\"auto\")")]
+    dtype: String,
+
+    /// device
+    #[argh(option, default = "String::from(\"cpu:0\")")]
+    device: String,
+
+    /// source image
+    #[argh(
+        option,
+        default = "vec![
+            String::from(\"images/dog.jpg\"),
+            String::from(\"images/siamese.png\"),
+            String::from(\"images/ailurus-fulgens.jpg\"),
+        ]"
+    )]
+    source: Vec<String>,
+}
+
+fn main() -> anyhow::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 = Options::mobileone_s0()
+        .with_model_dtype(args.dtype.as_str().try_into()?)
+        .with_model_device(args.device.as_str().try_into()?)
+        .commit()?;
+    let mut model = ImageClassifier::try_from(options)?;
+
+    // load images
+    let xs = DataLoader::try_read_batch(&args.source)?;
+
+    // run
+    let ys = model.forward(&xs)?;
+
+    // results
+    for (i, y) in ys.iter().enumerate() {
+        println!("{}: {:?}", i, y);
+    }
+
+    // annotate
+    let annotator = Annotator::default().with_saveout(model.spec());
+    annotator.annotate(&xs, &ys);
+
+    Ok(())
+}