diff --git a/mobilefacenet.py b/mobilefacenet.py index 571795b..1f95c26 100644 --- a/mobilefacenet.py +++ b/mobilefacenet.py @@ -49,7 +49,7 @@ class Mobilefacenet(NNOperator): normalize = transforms.Normalize( mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]) - self.tfms = transforms.Compose([transforms.Scale([112,112]), + self.tfms = transforms.Compose([transforms.Resize([112,112]), transforms.ToTensor(), normalize]) diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..09f3c27 --- /dev/null +++ b/requirements.txt @@ -0,0 +1,3 @@ +torch>=1.9.0 +torchvision>=0.10.0 +Pillow