|
|
|
@ -87,59 +87,59 @@ MTCNN主要有三个网络,叫做**PNet**, **RNet** 和 **ONet**。因此我
|
|
|
|
|
* 生成PNet训练数据和标注文件
|
|
|
|
|
|
|
|
|
|
```shell
|
|
|
|
|
python src/prepare_data/gen_Pnet_train_data.py --dataset_path {your dataset path} --anno_file {your dataset original annotation path}
|
|
|
|
|
python dface/prepare_data/gen_Pnet_train_data.py --dataset_path {your dataset path} --anno_file {your dataset original annotation path}
|
|
|
|
|
```
|
|
|
|
|
* 乱序合并标注文件
|
|
|
|
|
|
|
|
|
|
```shell
|
|
|
|
|
python src/prepare_data/assemble_pnet_imglist.py
|
|
|
|
|
python dface/prepare_data/assemble_pnet_imglist.py
|
|
|
|
|
```
|
|
|
|
|
|
|
|
|
|
* 训练PNet模型
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
```shell
|
|
|
|
|
python src/train_net/train_p_net.py
|
|
|
|
|
python dface/train_net/train_p_net.py
|
|
|
|
|
```
|
|
|
|
|
* 生成RNet训练数据和标注文件
|
|
|
|
|
|
|
|
|
|
```shell
|
|
|
|
|
python src/prepare_data/gen_Rnet_train_data.py --dataset_path {your dataset path} --anno_file {your dataset original annotation path} --pmodel_file {yout PNet model file trained before}
|
|
|
|
|
python dface/prepare_data/gen_Rnet_train_data.py --dataset_path {your dataset path} --anno_file {your dataset original annotation path} --pmodel_file {yout PNet model file trained before}
|
|
|
|
|
```
|
|
|
|
|
* 乱序合并标注文件
|
|
|
|
|
|
|
|
|
|
```shell
|
|
|
|
|
python src/prepare_data/assemble_rnet_imglist.py
|
|
|
|
|
python dface/prepare_data/assemble_rnet_imglist.py
|
|
|
|
|
```
|
|
|
|
|
|
|
|
|
|
* 训练RNet模型
|
|
|
|
|
|
|
|
|
|
```shell
|
|
|
|
|
python src/train_net/train_r_net.py
|
|
|
|
|
python dface/train_net/train_r_net.py
|
|
|
|
|
```
|
|
|
|
|
|
|
|
|
|
* 生成ONet训练数据和标注文件
|
|
|
|
|
|
|
|
|
|
```shell
|
|
|
|
|
python src/prepare_data/gen_Onet_train_data.py --dataset_path {your dataset path} --anno_file {your dataset original annotation path} --pmodel_file {yout PNet model file trained before} --rmodel_file {yout RNet model file trained before}
|
|
|
|
|
python dface/prepare_data/gen_Onet_train_data.py --dataset_path {your dataset path} --anno_file {your dataset original annotation path} --pmodel_file {yout PNet model file trained before} --rmodel_file {yout RNet model file trained before}
|
|
|
|
|
```
|
|
|
|
|
|
|
|
|
|
* 生成ONet的人脸关键点训练数据和标注文件
|
|
|
|
|
|
|
|
|
|
```shell
|
|
|
|
|
python src/prepare_data/gen_landmark_48.py
|
|
|
|
|
python dface/prepare_data/gen_landmark_48.py
|
|
|
|
|
```
|
|
|
|
|
|
|
|
|
|
* 乱序合并标注文件(包括人脸关键点)
|
|
|
|
|
|
|
|
|
|
```shell
|
|
|
|
|
python src/prepare_data/assemble_onet_imglist.py
|
|
|
|
|
python dface/prepare_data/assemble_onet_imglist.py
|
|
|
|
|
```
|
|
|
|
|
|
|
|
|
|
* 训练ONet模型
|
|
|
|
|
|
|
|
|
|
```shell
|
|
|
|
|
python src/train_net/train_o_net.py
|
|
|
|
|
python dface/train_net/train_o_net.py
|
|
|
|
|
```
|
|
|
|
|
|
|
|
|
|
#### 测试人脸检测
|
|
|
|
|