🐍 v0.1.0 (#53)

examples/dataloader/main.rs

@@ -1,66 +1,45 @@
-use usls::{
-    models::YOLO, Annotator, DataLoader, Device, Options, Viewer, Vision, YOLOTask, YOLOVersion,
-};
+use usls::DataLoader;
 
 fn main() -> anyhow::Result<()> {
     tracing_subscriber::fmt()
-        .with_max_level(tracing::Level::ERROR)
+        .with_env_filter(tracing_subscriber::EnvFilter::from_default_env())
+        .with_timer(tracing_subscriber::fmt::time::ChronoLocal::rfc_3339())
         .init();
 
-    let options = Options::new()
-        .with_device(Device::Cuda(0))
-        .with_model("yolo/v8-m-dyn.onnx")?
-        .with_yolo_version(YOLOVersion::V8)
-        .with_yolo_task(YOLOTask::Detect)
-        .with_batch(2)
-        .with_ixx(0, 2, (416, 640, 800).into())
-        .with_ixx(0, 3, (416, 640, 800).into())
-        .with_confs(&[0.2]);
-    let mut model = YOLO::new(options)?;
-
-    // build annotator
-    let annotator = Annotator::new()
-        .with_bboxes_thickness(4)
-        .with_saveout("YOLO-DataLoader");
-
-    // build dataloader
-    let dl = DataLoader::new(
+    // 1. iterator
+    let dl = DataLoader::try_from(
         // "images/bus.jpg", // remote image
         // "../images", // image folder
         // "../demo.mp4",   // local video
         // "http://commondatastorage.googleapis.com/gtv-videos-bucket/sample/BigBuckBunny.mp4", // remote video
         // "rtsp://admin:xyz@192.168.2.217:554/h265/ch1/",  // rtsp h264 stream
-        // "./assets/bus.jpg", // local image
-        "../7.mp4",
+        "./assets/bus.jpg", // local image
     )?
     .with_batch(1)
+    .with_progress_bar(true)
     .build()?;
 
-    let mut viewer = Viewer::new().with_delay(10).with_scale(1.).resizable(true);
-
-    // iteration
-    for (xs, _) in dl {
-        // inference & annotate
-        let ys = model.run(&xs)?;
-        let images_plotted = annotator.plot(&xs, &ys, false)?;
-
-        // show image
-        viewer.imshow(&images_plotted)?;
-
-        // check out window and key event
-        if !viewer.is_open() || viewer.is_key_pressed(usls::Key::Escape) {
-            break;
-        }
-
-        // write video
-        viewer.write_batch(&images_plotted)?;
+    for (_xs, _paths) in dl {
+        println!("Paths: {:?}", _paths);
     }
 
-    // finish video write
-    viewer.finish_write()?;
+    // 2. read one image
+    let image = DataLoader::try_read("./assets/bus.jpg")?;
+    println!(
+        "Read one image. Height: {}, Width: {}",
+        image.height(),
+        image.width()
+    );
 
-    // images -> video
-    // DataLoader::is2v("runs/YOLO-DataLoader", &["runs", "is2v"], 24)?;
+    // 3. read several images
+    let images = DataLoader::try_read_batch(&[
+        "./assets/bus.jpg",
+        "./assets/bus.jpg",
+        "./assets/bus.jpg",
+        "./assets/bus.jpg",
+        "./assets/bus.jpg",
+    ])?;
+    println!("Read {} images.", images.len());
 
     Ok(())
 }