0.1.0-beta.1 (#82)

examples/modnet/main.rs

@@ -11,14 +11,22 @@ fn main() -> anyhow::Result<()> {
     let mut model = MODNet::new(options)?;
 
     // load image
-    let xs = [DataLoader::try_read("images/liuyifei.png")?];
+    let xs = DataLoader::try_read_n(&["images/liuyifei.png"])?;
 
     // run
     let ys = model.forward(&xs)?;
 
     // annotate
-    let annotator = Annotator::default().with_saveout(model.spec());
-    annotator.annotate(&xs, &ys);
+    let annotator = Annotator::default();
+    for (x, y) in xs.iter().zip(ys.iter()) {
+        annotator.annotate(x, y)?.save(format!(
+            "{}.jpg",
+            usls::Dir::Current
+                .base_dir_with_subs(&["runs", model.spec()])?
+                .join(usls::timestamp(None))
+                .display(),
+        ))?;
+    }
 
     Ok(())
 }