Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
yolov5_migraphx
Commits
986bc897
Commit
986bc897
authored
Jun 13, 2023
by
shizhm
Browse files
修改readme
parent
8b5f8db4
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
14 additions
and
5 deletions
+14
-5
README.md
README.md
+14
-5
No files found.
README.md
View file @
986bc897
...
@@ -15,9 +15,9 @@ YoloV5模型的主要改进思路有以下几点:
...
@@ -15,9 +15,9 @@ YoloV5模型的主要改进思路有以下几点:
## python版本推理
## python版本推理
下面介绍如何运行
p
ython代码示例,
具体推理代码解析,在Doc/
Tutorial_Python.md
中有详细说明
。
下面介绍如何运行
P
ython代码示例,
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模型路径 \
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment