jackcc / RetinaFace-Cpp

RetinaFace detector with C++

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

RetinaFace-Cpp

RetinaFace detector with C++

official RetinaFace

I convert mobilenet-0.25 mxnet model (trained by yangfly) to caffe model

I convert R50 mxnet model to caffe model BaiDuYun | Google Drive

  • I have checked the output of the two models be the same.

  • For same input images, the output of the two detector (python version and cpp version) is same.

(the code is too simple, only for reference 23333)

This is anchor parsing demo, NO INFERENCE, you should replace your own inference header and code (caffe/ncnn/feather .etc) in the source code

I create QQ group: 760688309 for communicating


Update 2019.5.28

R50: Because of the GPU memory limited, I set max(width, height) to 1000 and test it on WiderFace_val set for SINGLE SCALE, NO MULTI-SCALE, NO FLIP, results as follows:

wider val easy medium hard
python version 92.25 88.86 64.02
cpp version 91.36 87.15 62.22

Update 2019.5.27

mnet: I test on WiderFace_val set for SINGLE SCALE, NO MULTI-SCALE, NO FLIP, results as follows:

wider val easy medium hard
python version 83.28 77.02 39.52
cpp version 83.04 76.84 39.43

You can see the two detect results as results/cpp_detect and results/python_detect. The main cause of the difference in test results is the numerical precision difference between C++ and python.

Time Benchmark

VGA (640x480)

platform net threads 1/2/4(ms) inference
qcom625 mnet 418/262/202 NCNN
qcom625 mnet 379/244/180 NCNN(optmize)
qcom835 mnet 130.94/-/- by hanson-young NCNN

TODO

  • implement bbox_vote
  • multi-scale test
  • speed benchmark (CPU/ARM)
  • new models

About

RetinaFace detector with C++


Languages

Language:C++ 100.0%