disabled window display

UBCAgroBot · Dec 7, 2024 · 10f340c · 10f340c
1 parent 3e8416e
commit 10f340c
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/ros2_ws/src/python_workspace/python_workspace/extermination_node.py b/ros2_ws/src/python_workspace/python_workspace/extermination_node.py
@@ -53,11 +53,11 @@ def inference_callback(self, msg):
         bounding_boxes = self.model.postprocess(msg.confidences.data,msg.bounding_boxes.data, raw_image,msg.velocity)
         # final_image = self.model.draw_boxes(raw_image,bounding_boxes,velocity=msg.velocity)
         labels = [f"{i}%" for i in msg.confidences.data]
-        final_image = display_annotated_image(raw_image, bounding_boxes, )
+        final_image = display_annotated_image(raw_image, bounding_boxes, labels)
 
-        if self.use_display_node:
-            cv2.imshow(self.window, final_image)
-            cv2.waitKey(0)
+        # if self.use_display_node:
+        #     cv2.imshow(self.window, final_image)
+        #     cv2.waitKey(0)
 
         if len(bounding_boxes) > 0:
             self.boxes_present = 1