1

.gitignore

@@ -3,4 +3,3 @@
 **/build/
 **/install/
 **/log/
-

vision_test/include/vision_test/ultralytics/get_masks.hpp

@@ -1,4 +1,13 @@
 
 
 
-void
+void seg_postprocess(
+    const float* predictions,
+    const float* proto,
+    int height,
+    int width,
+    int num_classes=80,
+    int mask_dim=32,
+    float conf_threshold=0.25,
+    float iou_threshold=0.25
+);

vision_test/src/ultralytics/get_masks.cpp

@@ -3,3 +3,11 @@
 #include "vision_test/ultralytics/get_masks.hpp"
 
 
+void seg_postprocess(
+    int num_classes,
+    int mask_dim,
+    float conf_threshold,
+    float iou_threshold
+) {
+    // Function implementation goes here
+}