Update eval/tools/face_id.py
Browse files- eval/tools/face_id.py +1 -0
eval/tools/face_id.py
CHANGED
@@ -51,6 +51,7 @@ class FaceID:
|
|
51 |
transforms.ToTensor(),
|
52 |
transforms.Normalize(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5])
|
53 |
])
|
|
|
54 |
self.detector = facer.face_detector("retinaface/resnet50", device=device)
|
55 |
face_model_path = os.getenv("FACE_ID_MODEL_PATH")
|
56 |
self.model = Backbone(num_layers=50, drop_ratio=0.6, mode='ir_se')
|
|
|
51 |
transforms.ToTensor(),
|
52 |
transforms.Normalize(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5])
|
53 |
])
|
54 |
+
|
55 |
self.detector = facer.face_detector("retinaface/resnet50", device=device)
|
56 |
face_model_path = os.getenv("FACE_ID_MODEL_PATH")
|
57 |
self.model = Backbone(num_layers=50, drop_ratio=0.6, mode='ir_se')
|