diff --git a/sam2/utils/transforms.py b/sam2/utils/transforms.py index 3a8ceaf..6482839 100644 --- a/sam2/utils/transforms.py +++ b/sam2/utils/transforms.py @@ -27,13 +27,23 @@ class SAM2Transforms(nn.Module): 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( - Resize((self.resolution, self.resolution)), - Normalize(self.mean, self.std), - ) + seq = nn.Sequential( + Resize((self.resolution, self.resolution)), + Normalize(self.mean, self.std), ) + try: + self.transforms = torch.jit.script(seq) + except Exception: + self.transforms = seq + + # self.transforms = torch.jit.script( + # nn.Sequential( + # Resize((self.resolution, self.resolution)), + # Normalize(self.mean, self.std), + # ) + # ) + def __call__(self, x): x = self.to_tensor(x) return self.transforms(x)