From 2f08306898384aec21329696b601af917a703a2c Mon Sep 17 00:00:00 2001 From: wxywb Date: Thu, 9 Jun 2022 10:48:33 +0800 Subject: [PATCH] add the requirement. Signed-off-by: wxywb --- mobilefacenet.py | 2 +- requirements.txt | 3 +++ 2 files changed, 4 insertions(+), 1 deletion(-) create mode 100644 requirements.txt 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