From c6328be105302943548bf3b5917b96446949143d Mon Sep 17 00:00:00 2001 From: Janette Rounds Date: Thu, 2 Jan 2025 17:53:25 -0700 Subject: [PATCH] undefined attribute in fast sam, fixed by setting the image variable (#384) Co-authored-by: Janette Rounds --- samgeo/fast_sam.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/samgeo/fast_sam.py b/samgeo/fast_sam.py index 5dfc9b54..1bdd41ef 100644 --- 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())