feat: Added nms

2025-08-05 15:36:47 +05:30
parent 42ac210bba
commit 06fb0b4487
5 changed files with 85 additions and 25 deletions
--- a/bounding-box/src/draw.rs
+++ b/bounding-box/src/draw.rs
@@ -57,12 +57,17 @@ impl Drawable<Array3<u8>> for Aabb2<usize> {
        lines.into_iter().for_each(|line| {
            canvas
                .roi_mut(line)
-                .expect("Failed to get Roi")
-                .lanes_mut(ndarray::Axis(2))
-                .into_iter()
-                .for_each(|mut pixel| {
-                    pixel.assign(&color);
-                });
+                .map(|mut line| {
+                    line.lanes_mut(ndarray::Axis(2))
+                        .into_iter()
+                        .for_each(|mut pixel| {
+                            pixel.assign(&color);
+                        })
+                })
+                .inspect_err(|e| {
+                    dbg!(e);
+                })
+                .ok();
        });
    }
 }