diff --git a/fluid/face_detection/README_cn.md b/fluid/face_detection/README_cn.md
index 8987b00fb2c66daedd42993214b5c9bab42a99c3..a160fbd9ce25c4fac4333b9914058a16fda7e58a 100644
--- a/fluid/face_detection/README_cn.md
+++ b/fluid/face_detection/README_cn.md
@@ -93,13 +93,14 @@ tar -xf vgg_ilsvrc_16_fc_reduced.tar.gz && rm -f vgg_ilsvrc_16_fc_reduced.tar.gz
`train.py` 是训练模块的主要执行程序,调用示例如下:
```bash
-python -u train.py --batch_size=12 --pretrained_model=vgg_ilsvrc_16_fc_reduced
+python -u train.py --batch_size=16 --pretrained_model=vgg_ilsvrc_16_fc_reduced
```
- - 可以通过设置 `export CUDA_VISIBLE_DEVICES=0,1,2,3` 指定想要使用的GPU数量。
+ - 可以通过设置 `export CUDA_VISIBLE_DEVICES=0,1,2,3` 指定想要使用的GPU数量,`batch_size`默认设置为12或16。
- 更多的可选参数见:
```bash
python train.py --help
```
+ - 模型训练150轮以上可以收敛。用Nvidia Tesla P40 GPU 4卡并行,`batch_size=16`的配置,每轮训练大约40分钟,总共训练时长大约100小时
模型训练所采用的数据增强:
@@ -152,8 +153,25 @@ python -u train.py --batch_size=12 --pretrained_model=vgg_ilsvrc_16_fc_reduced
```bash
matlab -nodesktop -nosplash -nojvm -r "run wider_eval.m;quit;"
+
```
+### 模型预测以及可视化
+`widerface_eval.py`也可以用来做预测及可视化,调用示例如下:
+
+```bash
+python widerface_eval.py --infer=True --confs_threshold=0.15
+ --model_dir=output/159/ --image_path=data/WIDER_train/images/0--Parade/0_Parade_marchingband_1_219.jpg
+```
+下图可视化了模型的预测结果:
+
+
+
+
+
+Pyramidbox 预测可视化
+
+
### 模型发布
diff --git a/fluid/face_detection/images/0_Parade_marchingband_1_356.jpg b/fluid/face_detection/images/0_Parade_marchingband_1_356.jpg
new file mode 100644
index 0000000000000000000000000000000000000000..d5175c5a9d57184945ae63ab1cfc0a2621ef9b70
Binary files /dev/null and b/fluid/face_detection/images/0_Parade_marchingband_1_356.jpg differ
diff --git a/fluid/face_detection/images/28_Sports_Fan_Sports_Fan_28_770.jpg b/fluid/face_detection/images/28_Sports_Fan_Sports_Fan_28_770.jpg
new file mode 100644
index 0000000000000000000000000000000000000000..661ad9abb2d21e6e6b872b2a44e14fc476e732e7
Binary files /dev/null and b/fluid/face_detection/images/28_Sports_Fan_Sports_Fan_28_770.jpg differ
diff --git a/fluid/face_detection/images/2_Demonstration_Demonstration_Or_Protest_2_58.jpg b/fluid/face_detection/images/2_Demonstration_Demonstration_Or_Protest_2_58.jpg
new file mode 100644
index 0000000000000000000000000000000000000000..febe7411ae2ff258a608275fc9050294014e8342
Binary files /dev/null and b/fluid/face_detection/images/2_Demonstration_Demonstration_Or_Protest_2_58.jpg differ
diff --git a/fluid/face_detection/images/4_Dancing_Dancing_4_194.jpg b/fluid/face_detection/images/4_Dancing_Dancing_4_194.jpg
new file mode 100644
index 0000000000000000000000000000000000000000..8161593708db628c6beb4f32c8133226e19f0f13
Binary files /dev/null and b/fluid/face_detection/images/4_Dancing_Dancing_4_194.jpg differ