提交 fcc6360a 编写于 作者: C channingss

update README

上级 69f7378d
...@@ -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,15 +50,22 @@ python infer.py --models_dir=/path/to/models --img_paths=/path/to/images --video ...@@ -50,15 +50,22 @@ 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/`下面。 执行完预测后,预测的可视化结果会保存在当前路径下的`./result/`下面。
输入样例: 输入样例:
![avatar](./images/mask.jpg) ![avatar](./images/mask.jpg)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册