logo
Browse Source

remove the transforms.Scale.

Signed-off-by: wxywb <xy.wang@zilliz.com>
main
wxywb 3 years ago
parent
commit
8b6fa07548
  1. 1
      dolg.py
  2. 1
      requirements.txt

1
dolg.py

@ -38,7 +38,6 @@ class Dolg(NNOperator):
normalize = transforms.Normalize(mean=[0.485, 0.456, 0.406],
std=[0.229, 0.224, 0.225])
self.tfms = transforms.Compose([transforms.Resize([img_size, img_size]),
transforms.Scale([img_size, img_size]),
transforms.ToTensor(),
normalize])

1
requirements.txt

@ -1,5 +1,4 @@
towhee
torch
torchvision
torch
timm

Loading…
Cancel
Save