未验证 提交 0d127b2b 编写于 作者: G Guanghua Yu 提交者: GitHub

fix python deploy --model_dir (#629)

上级 cb3875d4
......@@ -28,7 +28,7 @@ PaddleDetection在训练过程包括网络的前向和优化器相关参数,
在终端输入以下命令进行预测:
```bash
python infer.py --models_dir=/path/to/models --image_file=/path/to/image
python deploy/python/infer.py --model_dir=/path/to/models --image_file=/path/to/image
--use_gpu=(False/True)
```
......@@ -36,7 +36,7 @@ python infer.py --models_dir=/path/to/models --image_file=/path/to/image
| 参数 | 是否必须|含义 |
|-------|-------|----------|
| --models_dir | Yes|上述导出的模型路径 |
| --model_dir | Yes|上述导出的模型路径 |
| --image_file | Yes |需要预测的图片 |
| --video_file | Yes |需要预测的视频 |
| --use_gpu |No|是否GPU,默认为False|
......
......@@ -531,7 +531,7 @@ if __name__ == '__main__':
type=str,
default=None,
help=("Directory include:'__model__', '__params__', "
"'infer_cfg.yml', created by export_model."),
"'infer_cfg.yml', created by tools/export_model.py."),
required=True)
parser.add_argument(
"--image_file", type=str, default='', help="Path of image file.")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册