draw only centroid for rect2d

Author: SarangaR

src/vision/gate_poles.rs

@@ -2,7 +2,9 @@ use anyhow::Result;
 use derive_getters::Getters;
 use itertools::MergeJoinBy;
 use opencv::core::{multiply, multiply_def, MatTraitManual, BORDER_CONSTANT, CV_8U};
-use opencv::imgproc::{get_structuring_element, morphology_default_border_value, MORPH_RECT};
+use opencv::imgproc::{
+    dilate_def, get_structuring_element, morphology_default_border_value, MORPH_RECT,
+};
 use opencv::{
     core::{in_range, merge, split, Point, Scalar, Size, Vector},
     imgproc::{

src/vision/mod.rs

@@ -291,16 +291,20 @@ impl RelPos for DrawRect2d {
 
 impl Draw for DrawRect2d {
     fn draw(&self, canvas: &mut Mat) -> Result<()> {
-        imgproc::rectangle(
-            canvas,
-            self.inner
-                .to()
-                .ok_or(anyhow!("f64 outside bounds of i32"))?,
-            Scalar::from((0.0, 0.0, 255.0)),
-            2,
-            LINE_8,
-            0,
-        )?;
+        // imgproc::rectangle(
+        //     canvas,
+        //     self.inner
+        //         .to()
+        //         .ok_or(anyhow!("f64 outside bounds of i32"))?,
+        //     Scalar::from((0.0, 0.0, 255.0)),
+        //     2,
+        //     LINE_8,
+        //     0,
+        // )?;
+
+        let center = Point::new((self.x as i32), (self.y as i32));
+
+        imgproc::circle_def(canvas, center, 5, Scalar::from((0.0, 0.0, 255.0)))?;
         Ok(())
     }
 }

src/vision/yolo_model.rs

@@ -85,6 +85,7 @@ impl<T: Display> Draw for VisualDetection<YoloClass<T>, DrawRect2d> {
             LINE_AA,
             false,
         )?;
+
         Ok(())
     }
 }