diff --git a/samgeo/hq_sam.py b/samgeo/hq_sam.py index a0bfce68..784b6110 100644 --- 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 index 4e7fb244..03a6affb 100644 --- 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