1. 程式人生 > >基於Faster-RCNN的人臉檢測

基於Faster-RCNN的人臉檢測

文章主要使用Faster-RCNN框架,使用wider-face訓練資料集進行人臉檢測的訓練。VGG基礎框架的faceFaster-RCNN模型大小522M

安裝測試:

git clone --recursive https://github.com/playerkk/face-py-faster-rcnn.git
cd face-py-faster-rcnn
cd lib
make -j8
cd ..
cp  Makefile.config.example  Makefile.config

# In your Makefile.config, make sure to have this line uncommented
WITH_PYTHON_LAYER := 1
# Unrelatedly, it's also recommended that you use CUDNN
USE_CUDNN := 1

make all -j32
make pycaffe -j8

pip install protobuf==3.5.1

#download the pretrained face detector model
cd data/faster_rcnn_models
wget http://supermoe.cs.umass.edu/~hzjiang/data/vgg16_faster_rcnn_iter_80000.caffemodel

#test
python tools/demo.py

References:

http://download.csdn.net/download/qq_14845119/10249344