Merge pull request #885 from weecology/plot_results_flips_BGR

weecology · Jan 15, 2025 · dc51ef5 · dc51ef5
2 parents 7d0a0bb + 6d0c5c3
commit dc51ef5
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