undefined attribute in fast sam, fixed by setting the image variable

opengeos · Jan 2, 2025 · 071d3ed · 071d3ed
1 parent 956f644
commit 071d3ed
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/samgeo/fast_sam.py b/samgeo/fast_sam.py
@@ -227,9 +227,9 @@ def fast_show_mask(
         Returns:
             np.ndarray: The mask of the image.
         """
-
-        target_height = self.image.shape[0]
-        target_width = self.image.shape[1]
+        image = self.prompt_process.img
+        target_height = image.shape[0]
+        target_width = image.shape[1]
         annotations = self.annotations
         annotation = np.array(annotations.cpu())