Update eval/tools/face_id.py
Browse files- eval/tools/face_id.py +1 -0
eval/tools/face_id.py
CHANGED
@@ -54,6 +54,7 @@ class FaceID:
|
|
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')
|
58 |
self.model.load_state_dict(torch.load(face_model_path, map_location=device))
|
59 |
self.model.to(device)
|
|
|
54 |
|
55 |
self.detector = facer.face_detector("retinaface/resnet50", device=device)
|
56 |
face_model_path = os.getenv("FACE_ID_MODEL_PATH")
|
57 |
+
print(f'this is the way {face_model_path}')
|
58 |
self.model = Backbone(num_layers=50, drop_ratio=0.6, mode='ir_se')
|
59 |
self.model.load_state_dict(torch.load(face_model_path, map_location=device))
|
60 |
self.model.to(device)
|