Wwupup
commited on
Commit
·
60c396d
1
Parent(s):
4ae02a3
update yunet to v2 (#151)
Browse files
README.md
CHANGED
@@ -13,8 +13,8 @@ Results of accuracy evaluation with [tools/eval](../../tools/eval).
|
|
13 |
|
14 |
| Models | Easy AP | Medium AP | Hard AP |
|
15 |
| ----------- | ------- | --------- | ------- |
|
16 |
-
| YuNet | 0.
|
17 |
-
| YuNet quant | 0.
|
18 |
|
19 |
\*: 'quant' stands for 'quantized'.
|
20 |
|
|
|
13 |
|
14 |
| Models | Easy AP | Medium AP | Hard AP |
|
15 |
| ----------- | ------- | --------- | ------- |
|
16 |
+
| YuNet | 0.8871 | 0.8710 | 0.7681 |
|
17 |
+
| YuNet quant | 0.8838 | 0.8683 | 0.7676 |
|
18 |
|
19 |
\*: 'quant' stands for 'quantized'.
|
20 |
|
demo.cpp
CHANGED
@@ -112,7 +112,7 @@ int main(int argc, char** argv)
|
|
112 |
cv::CommandLineParser parser(argc, argv,
|
113 |
"{help h | | Print this message}"
|
114 |
"{input i | | Set input to a certain image, omit if using camera}"
|
115 |
-
"{model m |
|
116 |
"{backend b | opencv | Set DNN backend}"
|
117 |
"{target t | cpu | Set DNN target}"
|
118 |
"{save s | false | Whether to save result image or not}"
|
|
|
112 |
cv::CommandLineParser parser(argc, argv,
|
113 |
"{help h | | Print this message}"
|
114 |
"{input i | | Set input to a certain image, omit if using camera}"
|
115 |
+
"{model m | face_detection_yunet_2023mar.onnx | Set path to the model}"
|
116 |
"{backend b | opencv | Set DNN backend}"
|
117 |
"{target t | cpu | Set DNN target}"
|
118 |
"{save s | false | Whether to save result image or not}"
|
demo.py
CHANGED
@@ -27,8 +27,8 @@ backend_target_pairs = [
|
|
27 |
parser = argparse.ArgumentParser(description='YuNet: A Fast and Accurate CNN-based Face Detector (https://github.com/ShiqiYu/libfacedetection).')
|
28 |
parser.add_argument('--input', '-i', type=str,
|
29 |
help='Usage: Set input to a certain image, omit if using camera.')
|
30 |
-
parser.add_argument('--model', '-m', type=str, default='
|
31 |
-
help="Usage: Set model type, defaults to '
|
32 |
parser.add_argument('--backend_target', '-bt', type=int, default=0,
|
33 |
help='''Choose one of the backend-target pair to run this demo:
|
34 |
{:d}: (default) OpenCV implementation + CPU,
|
|
|
27 |
parser = argparse.ArgumentParser(description='YuNet: A Fast and Accurate CNN-based Face Detector (https://github.com/ShiqiYu/libfacedetection).')
|
28 |
parser.add_argument('--input', '-i', type=str,
|
29 |
help='Usage: Set input to a certain image, omit if using camera.')
|
30 |
+
parser.add_argument('--model', '-m', type=str, default='face_detection_yunet_2023mar.onnx',
|
31 |
+
help="Usage: Set model type, defaults to 'face_detection_yunet_2023mar.onnx'.")
|
32 |
parser.add_argument('--backend_target', '-bt', type=int, default=0,
|
33 |
help='''Choose one of the backend-target pair to run this demo:
|
34 |
{:d}: (default) OpenCV implementation + CPU,
|