diff --git a/configs/face_detection/README.md b/configs/face_detection/README.md index 5eb320f87bf18bd59df25e94b022a8e411f17a83..97adf2a306d72b3c123daf23be46e11f7a83705c 100644 --- a/configs/face_detection/README.md +++ b/configs/face_detection/README.md @@ -12,11 +12,10 @@ | 网络结构 | 输入尺寸 | 图片个数/GPU | 学习率策略 | Easy/Medium/Hard Set | 预测时延(SD855)| 模型大小(MB) | 下载 | 配置文件 | |:------------:|:--------:|:----:|:-------:|:-------:|:---------:|:----------:|:---------:|:--------:| | BlazeFace | 640 | 8 | 1000e | 0.885 / 0.855 / 0.731 | - | 0.472 |[下载链接](https://paddledet.bj.bcebos.com/models/blazeface_1000e.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/release/2.1/configs/face_detection/blazeface_1000e.yml) | -| BlazeFace-FPN-SSH | 640 | 8 | 1000e | 0.907 / 0.883 / 0.793 | - | 0.479 |[下载链接](https://paddledet.bj.bcebos.com/models/blazeface_fpn_ssh_1000e.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/release/2.1/configs/face_detection/blazeface_fpn_ssh_1000e.yml) | +| BlazeFace-FPN-SSH | 640 | 8 | 1000e | 0.920 / 0.900 / 0.822 | - | 0.646 |[下载链接](https://paddledet.bj.bcebos.com/models/blazeface_fpn_ssh_1000e.pdparams) | [配置文件](https://github.com/PaddlePaddle/PaddleDetection/tree/release/2.1/configs/face_detection/blazeface_fpn_ssh_1000e.yml) | **注意:** - 我们使用多尺度评估策略得到`Easy/Medium/Hard Set`里的mAP。具体细节请参考[在WIDER-FACE数据集上评估](#在WIDER-FACE数据集上评估)。 - ## 快速开始 ### 数据准备 @@ -79,9 +78,9 @@ BlazeNeck: ```shell python -u tools/eval.py -c configs/face_detection/blazeface_1000e.yml \ -o weights=output/blazeface_1000e/model_final \ - multi_scale=True + multi_scale_eval=True ``` -设置`multi_scale=True`进行多尺度评估,评估完成后,将在`output/pred`中生成txt格式的测试结果。 +设置`multi_scale_eval=True`进行多尺度评估,评估完成后,将在`output/pred`中生成txt格式的测试结果。 - 步骤二:下载官方评估脚本和Ground Truth文件: ``` @@ -112,6 +111,7 @@ matlab -nodesktop -nosplash -nojvm -r "run wider_eval.m;quit;" ``` + ## Citations ``` diff --git a/configs/face_detection/_base_/blazeface.yml b/configs/face_detection/_base_/blazeface.yml index de54100fe63c1d0dd004c5c1797b6a6587106993..5ac5069e7c7a4b2b292c57b2f39168be4cb893b3 100644 --- a/configs/face_detection/_base_/blazeface.yml +++ b/configs/face_detection/_base_/blazeface.yml @@ -1,4 +1,5 @@ architecture: BlazeFace +pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/blazenet_pretrain.pdparams BlazeFace: backbone: BlazeNet diff --git a/configs/face_detection/_base_/blazeface_fpn.yml b/configs/face_detection/_base_/blazeface_fpn.yml index 6572a99d301eda65a65c485e133cc00497a2eee2..1594b19a57024ca0aaff7aaa59d4337fd833edc0 100644 --- a/configs/face_detection/_base_/blazeface_fpn.yml +++ b/configs/face_detection/_base_/blazeface_fpn.yml @@ -1,4 +1,5 @@ architecture: BlazeFace +pretrain_weights: https://paddledet.bj.bcebos.com/models/pretrained/blazenet_pretrain.pdparams BlazeFace: backbone: BlazeNet