dlib cnn_face_detector 人脸检测内部实现简单记录
最近稍微研究了一下dlib人脸检测算法,在这里简单记录一下。
使用方法见 http://dlib.net/cnn_face_detector.py.html
- 前处理
dlib给的官方链接里面是这么使用的,参数列表里面的第二项数值1,表示放大2倍再塞给网络,不想放大的话,直接设为0就可以了
dets = cnn_face_detector(img, 1)
- 模型
dib的mmod的人脸检测模型是这样的。列的很清楚了,是通过卷积网络实现的。
1 // ---------------------------------------------------------------------------------------- 2 3 template <long num_filters, typename SUBNET> using con5d = con<num_filters,5,5,2,2,SUBNET>; 4 template <long num_filters, typename SUBNET> using con5 = con<num_filters,5,5,1,1,SUBNET>; 5 6 template <typename SUBNET> using downsampler = relu<affine<con5d<32, relu<affine<con5d<32, relu<affine<con5d<16,SUBNET>>>>>>>>>; 7 template <typename SUBNET> using rcon5 = relu<affine<con5<45,SUBNET>>>; 8 9 using net_type = loss_mmod<con<1,9,9,1,1,rcon5<rcon5<rcon5<downsampler<input_rgb_image_pyramid<pyramid_down<6>>>>>>>>; 10 11 // ----------------------------------------------------------------------------------------
几个点需要注意,
1.一上来就有一个pyramid_down 6就是把原图按照5/6的scaling ratio去缩小,形成图像金字塔,直到最小的图像的高度小于5
下图是一个Input为375x500的图,可以看到,一共有23层的图像金字塔,最小的一层是5x6. 这也就是dlib非常慢的原因。
2.其中的affine其实是caffe里面的scale,类似于BN,每个值放大a倍再加一个b。
- 后处理
在dlib内部会做nms,还会丢弃confidence threshould过低的结果,所以算出来的PR曲线后半截会没有掉