alexnasa commited on
Commit
f2a06a2
·
verified ·
1 Parent(s): 1968b4f

Update eval/tools/face_id.py

Browse files
Files changed (1) hide show
  1. 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')