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

…384) Co-authored-by: Janette Rounds <[email protected]>
opengeos · Jan 3, 2025 · c6328be · c6328be
1 parent 956f644
commit c6328be
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())