[FIX] Added missing conversion from BGR to RGB after reading an image…

… using cv2.imread (#171)
opengeos · Aug 22, 2023 · ee31add · ee31add
1 parent dd691e3
commit ee31add
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/samgeo/hq_sam.py b/samgeo/hq_sam.py
@@ -199,6 +199,7 @@ def generate(
                 )
 
             image = cv2.imread(source)
+            image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
         elif isinstance(source, np.ndarray):
             image = source
             source = None

diff --git a/samgeo/samgeo.py b/samgeo/samgeo.py
@@ -196,6 +196,7 @@ def generate(
                 )
 
             image = cv2.imread(source)
+            image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
         elif isinstance(source, np.ndarray):
             image = source
             source = None