Update transforms.py (#32)

Resolved SAM2 inference error reporting issue
neverbiasu · Dec 27, 2024 · f4a2b84 · f4a2b84
1 parent 81f1414
commit f4a2b84
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/sam2/utils/transforms.py b/sam2/utils/transforms.py
@@ -25,11 +25,9 @@ def __init__(
         self.mean = [0.485, 0.456, 0.406]
         self.std = [0.229, 0.224, 0.225]
         self.to_tensor = ToTensor()
-        self.transforms = torch.jit.script(
-            nn.Sequential(
+        self.transforms = nn.Sequential(
                 Resize((self.resolution, self.resolution)),
                 Normalize(self.mean, self.std),
-            )
         )
 
     def __call__(self, x):