Commit 986bc897 authored by shizhm's avatar shizhm
Browse files

修改readme

parent 8b5f8db4
......@@ -15,9 +15,9 @@ YoloV5模型的主要改进思路有以下几点:
## python版本推理
下面介绍如何运行python代码示例,具体推理代码解析,在Doc/Tutorial_Python.md中有详细说明
下面介绍如何运行Python代码示例,Python示例的详细说明见Doc目录下的Tutorial_Python.md。
### 拉取镜像
### 下载镜像
在光源可拉取推理的docker镜像,YoloV5工程推荐的镜像如下:
......@@ -25,9 +25,13 @@ YoloV5模型的主要改进思路有以下几点:
docker pull image.sourcefind.cn:5000/dcu/admin/base/custom:ort1.14.0_migraphx3.0.0-dtk22.10.1
```
### 推理示例
### 设置Python环境变量
YoloV5模型的推理示例程序是YoloV5_infer_migraphx.py,使用如下命令运行该推理示例:
```
export PYTHONPATH=/opt/dtk/lib:$PYTHONPATH
```
### 安装依赖
```
# 进入python示例目录
......@@ -35,8 +39,13 @@ cd <path_to_yolov5_migraphx>/Python
# 安装依赖
pip install -r requirements.txt
```
# 运行程序
### 推理示例
YoloV5模型的推理示例程序是YoloV5_infer_migraphx.py,在Python目录下使用如下命令运行该推理示例:
```
python YoloV5_infer_migraphx.py \
--imgpath 测试图像路径 \
--modelpath onnx模型路径 \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment