pyzed patch for roi calibrator, disable display

UBCAgroBot · Dec 7, 2024 · d5570b5 · d5570b5
1 parent 6460f90
commit d5570b5
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 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
@@ -55,9 +55,9 @@ def inference_callback(self, msg):
         labels = [f"{i}%" for i in msg.confidences.data]
         final_image = display_annotated_image(raw_image, bounding_boxes, labels)
 
-        if self.use_display_node:
-            cv2.imshow("left window", final_image)
-            cv2.waitKey(1)
+        # if self.use_display_node:
+        #     cv2.imshow("left window", final_image)
+        #     cv2.waitKey(1)
 
         if len(bounding_boxes) > 0:
             self.boxes_present = 1

diff --git a/tools/calibration/roi_calibration.py b/tools/calibration/roi_calibration.py
@@ -39,6 +39,7 @@ def initialize_image_source(source_type="static_image", image_path='C:/Users/ish
         return files
 
     elif source_type == "zed_single":
+        import pyzed.sl as sl
         init = sl.InitParameters()
         cam = sl.Camera()
         init.camera_resolution = sl.RESOLUTION.HD1080
@@ -53,6 +54,7 @@ def initialize_image_source(source_type="static_image", image_path='C:/Users/ish
         return cam
 
     elif source_type == "zed_double":
+        import pyzed.sl
         init = sl.InitParameters()
         cam1 = sl.Camera()
         cam2 = sl.Camera()
@@ -178,6 +180,7 @@ def onTrack6(val):
 
         elif source_type == "zed_single":
             image, velocity = retrieve_zed_image(cam, orientation)
+            velocity = 0
             if orientation == "left":
                 shifted_roi_x = roi_x + abs(velocity) * shift_constant
             else: