diff --git a/dolg.py b/dolg.py index d4806b7..98f5ebe 100644 --- a/dolg.py +++ b/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]) diff --git a/requirements.txt b/requirements.txt index 73d65df..e191bba 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,4 @@ towhee torch torchvision -torch timm