diff --git a/benchmark/config/face_recognition_sface.yaml b/benchmark/config/face_recognition_sface.yaml index 6bbda424..85585874 100644 --- a/benchmark/config/face_recognition_sface.yaml +++ b/benchmark/config/face_recognition_sface.yaml @@ -13,4 +13,4 @@ Benchmark: Model: name: "SFace" - modelPath: "models/face_recognition_sface/face_recognition_sface_2021sep.onnx" \ No newline at end of file + modelPath: "models/face_recognition_sface/face_recognition_sface_2021dec.onnx" \ No newline at end of file diff --git a/models/face_recognition_sface/demo.py b/models/face_recognition_sface/demo.py index 48b0e772..80259925 100644 --- a/models/face_recognition_sface/demo.py +++ b/models/face_recognition_sface/demo.py @@ -27,7 +27,7 @@ def str2bool(v): description="SFace: Sigmoid-Constrained Hypersphere Loss for Robust Face Recognition (https://ieeexplore.ieee.org/document/9318547)") parser.add_argument('--input1', '-i1', type=str, help='Path to the input image 1.') parser.add_argument('--input2', '-i2', type=str, help='Path to the input image 2.') -parser.add_argument('--model', '-m', type=str, default='face_recognition_sface_2021sep.onnx', help='Path to the model.') +parser.add_argument('--model', '-m', type=str, default='face_recognition_sface_2021dec.onnx', help='Path to the model.') parser.add_argument('--dis_type', type=int, choices=[0, 1], default=0, help='Distance type. \'0\': cosine, \'1\': norm_l1.') parser.add_argument('--save', '-s', type=str, default=False, help='Set true to save results. This flag is invalid when using camera.') 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): # Instantiate SFace for face recognition recognizer = SFace(modelPath=args.model, disType=args.dis_type) # Instantiate YuNet for face detection - detector = YuNet(modelPath='../face_detection_yunet/face_detection_yunet_2021sep.onnx', + detector = YuNet(modelPath='../face_detection_yunet/face_detection_yunet_2021dec.onnx', inputSize=[320, 320], confThreshold=0.9, nmsThreshold=0.3, diff --git a/models/face_recognition_sface/face_recognition_sface_2021dec.onnx b/models/face_recognition_sface/face_recognition_sface_2021dec.onnx new file mode 100644 index 00000000..5817e559 --- /dev/null +++ b/models/face_recognition_sface/face_recognition_sface_2021dec.onnx @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0ba9fbfa01b5270c96627c4ef784da859931e02f04419c829e83484087c34e79 +size 38696353 diff --git a/models/face_recognition_sface/face_recognition_sface_2021sep.onnx b/models/face_recognition_sface/face_recognition_sface_2021sep.onnx deleted file mode 100644 index 49f8a699..00000000 --- a/models/face_recognition_sface/face_recognition_sface_2021sep.onnx +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:a9279be00242f4240ba4de7b6b93837fdd5a17b98d2f278f63711d55b65b4312 -size 38696080