Fix mask bug in text_sam/predict when the input is PIL image (#117)

dvolgyes · web-flow · commit b82f2aeedd89 · 2023-06-21T07:06:30.000-04:00
diff --git a/samgeo/text_sam.py b/samgeo/text_sam.py
@@ -274,6 +274,7 @@ def predict(
                 )  # Convert numpy array to PIL image, excluding the alpha channel
         else:
             image_pil = image
+            image_np = np.array(image_pil)
 
         self.image = image_pil
 
