Remove torch.jit.script from SAM2Transforms class

main
Uminosachi 2024-08-06 19:02:54 +09:00
parent e54e285072
commit 247d388ad2
1 changed files with 3 additions and 5 deletions

View File

@ -25,12 +25,10 @@ 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(
self.transforms = nn.Sequential(
Resize((self.resolution, self.resolution), antialias=True),
Normalize(self.mean, self.std),
)
)
def __call__(self, x):
x = self.to_tensor(x)