From 8b6fa075483006e40b4e5f30aab1f9b6d59ad984 Mon Sep 17 00:00:00 2001 From: wxywb Date: Wed, 20 Jul 2022 17:02:13 +0800 Subject: [PATCH] remove the transforms.Scale. Signed-off-by: wxywb --- dolg.py | 1 - requirements.txt | 1 - 2 files changed, 2 deletions(-) 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