Skip to content

Commit bc04b1b

Browse files
authored
update sface: quantization ready (#35)
1 parent 8a872fb commit bc04b1b

File tree

4 files changed

+6
-6
lines changed

4 files changed

+6
-6
lines changed

benchmark/config/face_recognition_sface.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,4 +13,4 @@ Benchmark:
1313

1414
Model:
1515
name: "SFace"
16-
modelPath: "models/face_recognition_sface/face_recognition_sface_2021sep.onnx"
16+
modelPath: "models/face_recognition_sface/face_recognition_sface_2021dec.onnx"

models/face_recognition_sface/demo.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ def str2bool(v):
2727
description="SFace: Sigmoid-Constrained Hypersphere Loss for Robust Face Recognition (https://ieeexplore.ieee.org/document/9318547)")
2828
parser.add_argument('--input1', '-i1', type=str, help='Path to the input image 1.')
2929
parser.add_argument('--input2', '-i2', type=str, help='Path to the input image 2.')
30-
parser.add_argument('--model', '-m', type=str, default='face_recognition_sface_2021sep.onnx', help='Path to the model.')
30+
parser.add_argument('--model', '-m', type=str, default='face_recognition_sface_2021dec.onnx', help='Path to the model.')
3131
parser.add_argument('--dis_type', type=int, choices=[0, 1], default=0, help='Distance type. \'0\': cosine, \'1\': norm_l1.')
3232
parser.add_argument('--save', '-s', type=str, default=False, help='Set true to save results. This flag is invalid when using camera.')
3333
parser.add_argument('--vis', '-v', type=str2bool, default=True, help='Set true to open a window for result visualization. This flag is invalid when using camera.')
@@ -37,7 +37,7 @@ def str2bool(v):
3737
# Instantiate SFace for face recognition
3838
recognizer = SFace(modelPath=args.model, disType=args.dis_type)
3939
# Instantiate YuNet for face detection
40-
detector = YuNet(modelPath='../face_detection_yunet/face_detection_yunet_2021sep.onnx',
40+
detector = YuNet(modelPath='../face_detection_yunet/face_detection_yunet_2021dec.onnx',
4141
inputSize=[320, 320],
4242
confThreshold=0.9,
4343
nmsThreshold=0.3,
Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
version https://git-lfs.github.com/spec/v1
2+
oid sha256:0ba9fbfa01b5270c96627c4ef784da859931e02f04419c829e83484087c34e79
3+
size 38696353

models/face_recognition_sface/face_recognition_sface_2021sep.onnx

Lines changed: 0 additions & 3 deletions
This file was deleted.

0 commit comments

Comments
 (0)