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
yolov9_migraphx
Commits
d917ebc0
Commit
d917ebc0
authored
Aug 22, 2024
by
yaoht
Browse files
更新docker镜像以及相关说明
parent
c704fcfb
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
21 additions
and
3 deletions
+21
-3
CMakeLists.txt
CMakeLists.txt
+1
-1
README.md
README.md
+19
-1
docker/Dockerfile
docker/Dockerfile
+1
-1
No files found.
CMakeLists.txt
View file @
d917ebc0
...
...
@@ -17,7 +17,7 @@ set(INCLUDE_PATH ${CMAKE_CURRENT_SOURCE_DIR}/Src/
include_directories
(
${
INCLUDE_PATH
}
)
# 添加依赖库路径
set
(
LIBRARY_PATH
${
CMAKE_CURRENT_SOURCE_DIR
}
/depend/lib
64
/
set
(
LIBRARY_PATH
${
CMAKE_CURRENT_SOURCE_DIR
}
/depend/lib/
$ENV{DTKROOT}/lib/
)
link_directories
(
${
LIBRARY_PATH
}
)
...
...
README.md
View file @
d917ebc0
...
...
@@ -31,7 +31,7 @@ YOLOv9提出了新网络架构——GELAN。GELAN通过结合两种神经网络
拉取镜像:
```
shell
docker pull image.sourcefind.cn:5000/dcu/admin/base/migraphx:4.
2.1-centos7.6
-dtk24.04-py310
docker pull image.sourcefind.cn:5000/dcu/admin/base/migraphx:4.
3.0-ubuntu20.04
-dtk24.04
.1
-py3
.
10
```
创建并启动容器:
...
...
@@ -101,6 +101,16 @@ python YoloV9_infer_migraphx.py --dynamicInfer
### C++版本推理
注意:当使用操作系统不一样时,CMakeList需要做相应的修改:
```
# ubuntu操作系统
${CMAKE_CURRENT_SOURCE_DIR}/depend/lib64/ 修改为 ${CMAKE_CURRENT_SOURCE_DIR}/depend/lib/
# centos操作系统
${CMAKE_CURRENT_SOURCE_DIR}/depend/lib/ 修改为 ${CMAKE_CURRENT_SOURCE_DIR}/depend/lib64/
```
下面介绍如何运行C++代码示例,C++示例的详细说明见Doc目录下的Tutorial_Cpp.md。
...
...
@@ -115,6 +125,14 @@ rbuild build -d depend
将依赖库依赖加入环境变量LD_LIBRARY_PATH,在~/.bashrc中添加如下语句:
当操作系统是ubuntu系统时:
```
shell
export
LD_LIBRARY_PATH
=
<path_to_yolov9_migraphx>/depend/lib/:
$LD_LIBRARY_PATH
```
当操作系统是centos系统时:
```
shell
export
LD_LIBRARY_PATH
=
<path_to_yolov9_migraphx>/depend/lib64/:
$LD_LIBRARY_PATH
```
...
...
docker/Dockerfile
View file @
d917ebc0
FROM
image.sourcefind.cn:5000/dcu/admin/base/migraphx:4.
2.1-centos7.6
-dtk24.04-py310
FROM
image.sourcefind.cn:5000/dcu/admin/base/migraphx:4.
3.0-ubuntu20.04
-dtk24.04
.1
-py3
.
10
RUN
source
/opt/dtk/env.sh
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