Remove torch.jit.script from SAM2Transforms class
parent
e54e285072
commit
247d388ad2
|
|
@ -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)
|
||||
|
|
|
|||
Loading…
Reference in New Issue