init project

modules/mobilesamv2/modeling/sam.py

@@ -157,7 +157,7 @@ class Sam(nn.Module):
         """
         masks = F.interpolate(
             masks,
-            (self.image_encoder.img_size, self.image_encoder.img_size),
+            (self.image_encoder.image_size, self.image_encoder.image_size),
             mode="bilinear",
             align_corners=False,
         )
@@ -173,8 +173,8 @@ class Sam(nn.Module):
 
         # Pad
         h, w = x.shape[-2:]
-        padh = self.image_encoder.img_size - h
-        padw = self.image_encoder.img_size - w
+        padh = self.image_encoder.image_size - h
+        padw = self.image_encoder.image_size - w
         x = F.pad(x, (0, padw, 0, padh))
         return x
     
@@ -185,8 +185,8 @@ class Sam(nn.Module):
 
         # Pad
         h, w = x.shape[-2:]
-        padh = self.image_encoder.img_size - h
-        padw = self.image_encoder.img_size - w
+        padh = self.image_encoder.image_size - h
+        padw = self.image_encoder.image_size - w
         x = F.pad(x, (0, padw, 0, padh))
         return x
 
@@ -197,7 +197,7 @@ class Sam(nn.Module):
 
         # Pad
         # h, w = x.shape[-2:]
-        # padh = self.image_encoder.img_size - h
-        # padw = self.image_encoder.img_size - w
+        # padh = self.image_encoder.image_size - h
+        # padw = self.image_encoder.image_size - w
         # x = F.pad(x, (0, padw, 0, padh))
         return x