lukemelas commited on
Commit
c7054d6
1 Parent(s): cd30264

Update with larger max size

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -50,7 +50,7 @@ def get_transform(name: str):
50
  if any(x in name for x in ('dino', 'mocov3', 'convnext', )):
51
  normalize = transforms.Normalize((0.485, 0.456, 0.406), (0.229, 0.224, 0.225))
52
  transform = transforms.Compose([
53
- transforms.Resize(size=256, interpolation=TF.InterpolationMode.BICUBIC, max_size=384),
54
  transforms.ToTensor(),
55
  normalize
56
  ])
 
50
  if any(x in name for x in ('dino', 'mocov3', 'convnext', )):
51
  normalize = transforms.Normalize((0.485, 0.456, 0.406), (0.229, 0.224, 0.225))
52
  transform = transforms.Compose([
53
+ transforms.Resize(size=512, interpolation=TF.InterpolationMode.BICUBIC, max_size=1024),
54
  transforms.ToTensor(),
55
  normalize
56
  ])