基于PaddlePaddle复现的MTCNN人脸检测模型
APACHE-2.0 License
MTCNNMulti-task convolutional neural networkP-NetR-NetO-Net2016P-NetR-NetO-Net
models/Loss.py
MTCNNmodels/PNet.py
PNetmodels/RNet.py
RNetmodels/ONet.py
ONetutils/data_format_converter.py
utils/data.py
utils/utils.py
train_PNet/generate_PNet_data.py
PNettrain_PNet/train_PNet.py
PNettrain_RNet/generate_RNet_data.py
RNettrain_RNet/train_RNet.py
RNettrain_ONet/generate_ONet_data.py
ONettrain_ONet/train_ONet.py
ONetinfer_path.py
infer_camera.py
dataset``lfw_5590``net_7876``WIDER_train``testImageList.txt``trainImageList.txt``wider_face_train.txt
PNetRNetONet
PNetProposal NetworkP-Net
cd train_PNet
train_PNet
python3 generate_PNet_data.py
PNetpython3 train_PNet.py
PNetRefine NetworkP-NetP-NetR-NetBounding-Box RegressionNMS
cd train_RNet
train_RNet
python3 generate_RNet_data.py
PNetRNetpython3 train_RNet.py
RNetONetOutput NetworkR-NetO-NetR-Net
cd train_ONet
train_ONet
python3 generate_ONet_data.py
PNetRNetONetpython3 train_ONet.py
ONetpython3 infer_path.py
box
python3 infer_camera.py
box