更新docker镜像以及相关说明

CMakeLists.txt

@@ -17,7 +17,7 @@ set(INCLUDE_PATH    ${CMAKE_CURRENT_SOURCE_DIR}/Src/
 include_directories(${INCLUDE_PATH})
 
 # 添加依赖库路径
-set(LIBRARY_PATH ${CMAKE_CURRENT_SOURCE_DIR}/depend/lib64/
+set(LIBRARY_PATH ${CMAKE_CURRENT_SOURCE_DIR}/depend/lib/
                 $ENV{DTKROOT}/lib/)
 link_directories(${LIBRARY_PATH})
 

README.md

@@ -25,13 +25,13 @@ YOLOv7的作者提出了 Extended-ELAN (E-ELAN)结构。E-ELAN采用了ELAN类
 拉取镜像：
 
 ```plaintext
-docker pull image.sourcefind.cn:5000/dcu/admin/base/migraphx:4.0.0-centos7.6-dtk23.04.1-py38-latest
+docker pull image.sourcefind.cn:5000/dcu/admin/base/migraphx:4.3.0-ubuntu20.04-dtk24.04.1-py3.10
 ```
 
 创建并启动容器：
 
 ```plaintext
-docker run --shm-size 16g --network=host --name=yolov7_migraphx --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v $PWD/yolov7_migraphx:/home/yolov7_migraphx -it <Your Image ID> /bin/bash
+docker run --shm-size 16g --network=host --name=yolov7_migraphx --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v /opt/hyhal:/opt/hyhal:ro -v $PWD/yolov7_migraphx:/home/yolov7_migraphx -it <Your Image ID> /bin/bash
 
 # 激活dtk
 source /opt/dtk/env.sh
@@ -43,7 +43,7 @@ source /opt/dtk/env.sh
 cd ./docker
 docker build --no-cache -t yolov7_migraphx:2.0 .
 
-docker run --shm-size 16g --network=host --name=yolov7_migraphx --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v $PWD/yolov7_migraphx:/home/yolov7_migraphx -it <Your Image ID> /bin/bash
+docker run --shm-size 16g --network=host --name=yolov7_migraphx --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v /opt/hyhal:/opt/hyhal:ro -v $PWD/yolov7_migraphx:/home/yolov7_migraphx -it <Your Image ID> /bin/bash
 
 # 激活dtk
 source /opt/dtk/env.sh
@@ -90,6 +90,16 @@ python YoloV7_infer_migraphx.py \
 
 ### 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。
 
 
@@ -103,8 +113,16 @@ rbuild build -d depend
 
 将依赖库依赖加入环境变量LD_LIBRARY_PATH，在~/.bashrc中添加如下语句：
 
+当操作系统是ubuntu系统时：
+
+```shell
+export LD_LIBRARY_PATH=<path_to_yolov9_migraphx>/depend/lib/:$LD_LIBRARY_PATH
 ```
-export LD_LIBRARY_PATH=<path_to_yolov7_migraphx>/depend/lib64/:$LD_LIBRARY_PATH
+
+当操作系统是centos系统时：
+
+```shell
+export LD_LIBRARY_PATH=<path_to_yolov9_migraphx>/depend/lib64/:$LD_LIBRARY_PATH
 ```
 
 然后执行:

docker/Dockerfile

-FROM image.sourcefind.cn:5000/dcu/admin/base/migraphx:4.0.0-centos7.6-dtk23.04.1-py38-latest
\ No newline at end of file
+FROM image.sourcefind.cn:5000/dcu/admin/base/migraphx:4.3.0-ubuntu20.04-dtk24.04.1-py3.10
\ No newline at end of file