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
yolov7_migraphx
Commits
987611bc
Commit
987611bc
authored
Jun 13, 2023
by
shizhm
Browse files
修改readme
parent
37434443
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
24 additions
and
9 deletions
+24
-9
README.md
README.md
+24
-9
No files found.
README.md
View file @
987611bc
...
...
@@ -8,11 +8,11 @@ YOLOV7是2022年最新出现的一种YOLO系列目标检测模型,在论文 [Y
YoloV7模型的网络结构包括三个部分:input、backbone和head。与yolov5不同的是,将neck层与head层合称为head层,实际上的功能是一样的。各个部分的功能和yolov5相同,如backbone用于提取特征,head用于预测。yolov7依旧基于anchor based的方法,同时在网络架构上增加E-ELAN层,并将REP层也加入进来,方便后续部署,同时在训练时,在head时,新增Aux_detect用于辅助检测。
##
p
ython版本推理
##
P
ython版本推理
下面介绍如何运行
p
ython代码示例,
具体推理代码解析,在Doc/
Tutorial_Python.md
中有详细说明
。
下面介绍如何运行
P
ython代码示例,
Python示例的详细说明见Doc目录下的
Tutorial_Python.md。
###
拉取
镜像
###
下载
镜像
在光源可拉取推理的docker镜像,YoloV7工程推荐的镜像如下:
...
...
@@ -20,9 +20,13 @@ YoloV7模型的网络结构包括三个部分:input、backbone和head。与yol
docker
pull
image
.
sourcefind
.
cn
:
5000
/
dcu
/
admin
/
base
/
custom
:
ort1
.
14.0
_migraphx3
.
0.0
-
dtk22
.
10.1
```
###
推理示例
###
设置Python环境变量
YoloV7模型的推理示例程序是YoloV7_infer_migraphx.py,使用如下命令运行该推理示例:
```
export PYTHONPATH=/opt/dtk/lib:$PYTHONPATH
```
### 安装依赖
```
# 进入python示例目录
...
...
@@ -30,8 +34,13 @@ cd <path_to_yolov7_migraphx>/Python
# 安装依赖
pip install -r requirements.txt
```
### 运行示例
YoloV7模型的推理示例程序是YoloV7_infer_migraphx.py,在Python目录下使用如下命令运行该推理示例:
# 运行程序
```
python YoloV7_infer_migraphx.py \
--imgpath 测试图像路径 \
--modelpath onnx模型路径 \
...
...
@@ -46,9 +55,15 @@ python YoloV7_infer_migraphx.py \
## C++版本推理
下面介绍如何运行C++代码示例,具体推理代码解析,在Doc/Tutorial_Cpp.md目录中有详细说明。
下面介绍如何运行C++代码示例,C++示例的详细说明见Doc目录下的Tutorial_Cpp.md。
### 下载镜像
在光源中下载MIGraphX镜像:
参考Python版本推理中的构建安装,在光源中拉取推理的docker镜像。
```
docker pull image.sourcefind.cn:5000/dcu/admin/base/custom:ort1.14.0_migraphx3.0.0-dtk22.10.1
```
### 安装Opencv依赖
...
...
@@ -93,7 +108,7 @@ export LD_LIBRARY_PATH=<path_to_yolov7_migraphx>/depend/lib/:$LD_LIBRARY_PATH
source ~/.bashrc
```
###
推理
示例
###
运行
示例
成功编译YoloV7工程后,执行如下命令运行该示例:
...
...
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