未验证 提交 797445e5 编写于 作者: Z Zeyu Chen 提交者: GitHub

Add mask_detection python deploy README

update README.md
...@@ -15,7 +15,7 @@ ...@@ -15,7 +15,7 @@
请点击[官方安装文档](https://paddlepaddle.org.cn/install/quick) 选择适合的方式 请点击[官方安装文档](https://paddlepaddle.org.cn/install/quick) 选择适合的方式
- `PaddleHub`的安装: `pip install paddlehub` - `PaddleHub`的安装: `pip install paddlehub`
- `opencv`的安装: `pip install opencv-python` - `opencv`的安装: `pip install opencv-python`
-
#### 1.2 从`PaddleHub`导出预测模型 #### 1.2 从`PaddleHub`导出预测模型
``` ```
git clone https://github.com/PaddlePaddle/PaddleHub.git git clone https://github.com/PaddlePaddle/PaddleHub.git
...@@ -27,7 +27,7 @@ python export_model.py ...@@ -27,7 +27,7 @@ python export_model.py
`pyramidbox_lite_server_mask`为服务器版模型,在此推荐该版本模型,精度相对移动版本更高。 `pyramidbox_lite_server_mask`为服务器版模型,在此推荐该版本模型,精度相对移动版本更高。
成功执行代码后导出的模型路径结构: 成功执行代码后导出的模型路径结构:
``` ```
pyramidbox_lite_mobile_mask pyramidbox_lite_server_mask
| |
├── mask_detector # 口罩人脸分类模型 ├── mask_detector # 口罩人脸分类模型
| ├── __model__ # 模型文件 | ├── __model__ # 模型文件
...@@ -50,13 +50,24 @@ python infer.py --models_dir=/path/to/models --img_paths=/path/to/images --video ...@@ -50,13 +50,24 @@ python infer.py --models_dir=/path/to/models --img_paths=/path/to/images --video
| 参数 | 是否必须|含义 | | 参数 | 是否必须|含义 |
|-------|-------|----------| |-------|-------|----------|
| models_dir | Yes|两个模型路径./pyramidbox_lite_mobile_mask | | models_dir | Yes|上述导出的模型路径 |
| img_paths |img_paths/video_path 二选一|需要预测的图片目录 | | img_paths |img_paths/video_path 二选一|需要预测的图片目录 |
| video_path |img_paths/video_path 二选一|需要预测的视频目录| | video_path |img_paths/video_path 二选一|需要预测的视频目录|
| video_size |No|预测视频分辨率大小(w,h) | | video_size |No|预测视频分辨率大小(w,h) |
| use_camera |No|是否打开摄像头进行预测 | | use_camera |No|是否打开摄像头进行预测 |
| use_gpu |No|是否GPU,默认为False| | use_gpu |No|是否GPU,默认为False|
说明:
如果use_gpu=True,请先在命令行指定GPU,如:
```
export CUDA_VISIBLE_DEVICES=0
```
##3. 可视化 ## 3. 可视化结果
执行完预测后,预测的可视化结果会保存在当前路径下的`./result/`下面。
输入样例:
![avatar](./images/mask.jpg)
输出结果:
![avatar](./images/mask.jpg.result.jpg)
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册