plot results was using cv2 needs to be BGR order:

weecology · Jan 14, 2025 · 6d0c5c3 · 6d0c5c3
1 parent 7d0a0bb
commit 6d0c5c3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/deepforest/visualize.py b/src/deepforest/visualize.py
@@ -532,6 +532,8 @@ def plot_results(results,
                 results.image_path.unique()[0]))[0]
         image_name = "{}.png".format(basename)
         image_path = os.path.join(savedir, image_name)
+        # Flip RGB to BGR
+        annotated_scene = annotated_scene[:, :, ::-1]
         cv2.imwrite(image_path, annotated_scene)
     else:
         # Display the image using Matplotlib