Skip to content

Commit caed3dc

Browse files
authored
Update inference.py
1 parent 45a18a5 commit caed3dc

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

inference.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@ def main(params):
6464

6565
idx_tensor = torch.arange(params.bins, device=device, dtype=torch.float32)
6666

67-
face_detector = uniface.RetinaFace(model="retinaface_mnet_v2") # third-party face detection library
67+
face_detector = uniface.RetinaFace() # third-party face detection library
6868

6969
try:
7070
gaze_detector = get_model(params.model, params.bins, inference_mode=True)

0 commit comments

Comments
 (0)