更新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

@@ -27,13 +27,13 @@ YOLOv8算法通过将图像划分为不同大小的网格，预测每个网格
 拉取镜像：
 
 ```shell
-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
 ```
 
 创建并启动容器：
 
 ```shell
-docker run --shm-size 16g --network=host --name=yolov8_migraphx --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v $PWD/yolov8_migraphx:/home/yolov8_migraphx -it <Your Image ID> /bin/bash
+docker run --shm-size 16g --network=host --name=yolov8_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/yolov8_migraphx:/home/yolov8_migraphx -it <Your Image ID> /bin/bash
 
 # 激活dtk
 source /opt/dtk/env.sh
@@ -45,7 +45,7 @@ source /opt/dtk/env.sh
 cd ./docker
 docker build --no-cache -t yolov8_migraphx:2.0 .
 
-docker run --shm-size 16g --network=host --name=yolov8_migraphx --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v $PWD/yolov8_migraphx:/home/yolov8_migraphx -it <Your Image ID> /bin/bash
+docker run --shm-size 16g --network=host --name=yolov8_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/yolov8_migraphx:/home/yolov8_migraphx -it <Your Image ID> /bin/bash
 
 # 激活dtk
 source /opt/dtk/env.sh
@@ -96,6 +96,16 @@ python YoloV8_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。
 
 
@@ -109,8 +119,16 @@ 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_yolov8_migraphx>/depend/lib64/:$LD_LIBRARY_PATH
+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
+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