logo
Browse Source

add the requirement.

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

2
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])

3
requirements.txt

@ -0,0 +1,3 @@
torch>=1.9.0
torchvision>=0.10.0
Pillow
Loading…
Cancel
Save