Wwupup commited on
Commit
60c396d
·
1 Parent(s): 4ae02a3

update yunet to v2 (#151)

Browse files
Files changed (3) hide show
  1. README.md +2 -2
  2. demo.cpp +1 -1
  3. demo.py +2 -2
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.8498 | 0.8384 | 0.7357 |
17
- | YuNet quant | 0.7751 | 0.8145 | 0.7312 |
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 | face_detection_yunet_2022mar.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}"
 
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='face_detection_yunet_2022mar.onnx',
31
- help="Usage: Set model type, defaults to 'face_detection_yunet_2022mar.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,
 
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,