output_image

UBCAgroBot · Dec 1, 2024 · 07ede30 · 07ede30
1 parent 5db971d
commit 07ede30
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/ros2_ws/src/python_workspace/python_workspace/scripts/display.py b/ros2_ws/src/python_workspace/python_workspace/scripts/display.py
@@ -37,18 +37,19 @@ def main():
     names = {0: "Class A", 1: "Class B"}  # Class names
 
     # Plot the images with bounding boxes
-    plotting.plot_images(
+    output_image = plotting.plot_images(
         images=images,
         batch_idx=batch_idx,
         cls=cls,
         bboxes=bboxes,
         confs=confs,
         fname="output.jpg",
         names=names,
+        save=False
     )
 
     # Display the result using OpenCV
-    output_image = cv2.imread("output.jpg")
+    # output_image = cv2.imread("output.jpg")
     if output_image is not None:
         cv2.imshow("Image with Bounding Boxes", output_image)
         cv2.waitKey(0)