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

修改readme

parent 8b5f8db4
...@@ -15,9 +15,9 @@ YoloV5模型的主要改进思路有以下几点: ...@@ -15,9 +15,9 @@ YoloV5模型的主要改进思路有以下几点:
## python版本推理 ## python版本推理
下面介绍如何运行python代码示例,具体推理代码解析,在Doc/Tutorial_Python.md中有详细说明 下面介绍如何运行Python代码示例,Python示例的详细说明见Doc目录下的Tutorial_Python.md。
### 拉取镜像 ### 下载镜像
在光源可拉取推理的docker镜像,YoloV5工程推荐的镜像如下: 在光源可拉取推理的docker镜像,YoloV5工程推荐的镜像如下:
...@@ -25,9 +25,13 @@ 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 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示例目录 # 进入python示例目录
...@@ -35,8 +39,13 @@ cd <path_to_yolov5_migraphx>/Python ...@@ -35,8 +39,13 @@ cd <path_to_yolov5_migraphx>/Python
# 安装依赖 # 安装依赖
pip install -r requirements.txt pip install -r requirements.txt
```
# 运行程序 ### 推理示例
YoloV5模型的推理示例程序是YoloV5_infer_migraphx.py,在Python目录下使用如下命令运行该推理示例:
```
python YoloV5_infer_migraphx.py \ python YoloV5_infer_migraphx.py \
--imgpath 测试图像路径 \ --imgpath 测试图像路径 \
--modelpath onnx模型路径 \ --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