Commit 5d519488 authored by liucong's avatar liucong
Browse files

修改readme文档

parent 82baf0df
...@@ -8,34 +8,38 @@ U-net模型整体为U型结构,主要分为三部分:下采样、上采样 ...@@ -8,34 +8,38 @@ U-net模型整体为U型结构,主要分为三部分:下采样、上采样
## Python版本推理 ## Python版本推理
下面介绍如何运行python代码示例,具体推理代码解析,在Doc/Tutorial_Python.md中有详细说明 下面介绍如何运行python代码示例,Python示例的详细说明见Doc目录下的Tutorial_Python.md。
### 拉取镜像 ### 下载镜像
在光源可拉取推理的docker镜像,Unet模型推理的镜像如下 在光源中下载MIGraphX镜像
```python ```python
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环境变量
1.参考《MIGraphX教程》设置好PYTHONPATH ```
export PYTHONPATH=/opt/dtk/lib:$PYTHONPATH
```
2.安装依赖 ### 安装依赖
```Python ```Python
# 进入unet migraphx工程根目录 # 进入unet migraphx工程根目录
cd <path_to_unet_migraphx> cd <path_to_unet_migraphx>
# 进入示例程序目录 # 进入示例程序目录
cd ./Python/ cd Python/
# 安装依赖 # 安装依赖
pip install -r requirements.txt pip install -r requirements.txt
``` ```
3.在Python目录下执行如下命令运行该示例程序: ### 运行示例
在Python目录下执行如下命令运行该示例程序:
```python ```python
python Unet.py python Unet.py
...@@ -47,9 +51,15 @@ python Unet.py ...@@ -47,9 +51,15 @@ python Unet.py
## C++版本推理 ## C++版本推理
下面介绍如何运行C++代码示例,具体推理代码解析,在Doc/Tutorial_Cpp.md目录中有详细说明 下面介绍如何运行C++代码示例,C++示例的详细说明见Doc目录下的Tutorial_Cpp.md。
参考Python版本推理中的构建安装,在光源中拉取推理的docker镜像。 ### 下载镜像
在光源中下载MIGraphX镜像:
```
docker pull image.sourcefind.cn:5000/dcu/admin/base/custom:ort1.14.0_migraphx3.0.0-dtk22.10.1
```
### 安装Opencv依赖 ### 安装Opencv依赖
...@@ -60,10 +70,8 @@ sh ./3rdParty/InstallOpenCVDependences.sh ...@@ -60,10 +70,8 @@ sh ./3rdParty/InstallOpenCVDependences.sh
### 修改CMakeLists.txt ### 修改CMakeLists.txt
- 如果使用ubuntu系统,需要修改CMakeLists.txt中依赖库路径: 如果使用ubuntu系统,需要修改CMakeLists.txt中依赖库路径:
将"${CMAKE_CURRENT_SOURCE_DIR}/depend/lib64/"修改为"${CMAKE_CURRENT_SOURCE_DIR}/depend/lib/" 将"${CMAKE_CURRENT_SOURCE_DIR}/depend/lib64/"修改为"${CMAKE_CURRENT_SOURCE_DIR}/depend/lib/"
- **MIGraphX2.3.0及以上版本需要c++17**
### 安装OpenCV并构建工程 ### 安装OpenCV并构建工程
...@@ -94,9 +102,7 @@ export LD_LIBRARY_PATH=<path_to_unet_migraphx>/depend/lib/:$LD_LIBRARY_PATH ...@@ -94,9 +102,7 @@ export LD_LIBRARY_PATH=<path_to_unet_migraphx>/depend/lib/:$LD_LIBRARY_PATH
source ~/.bashrc source ~/.bashrc
``` ```
### 推理示例 ### 运行示例
运行Unet示例程序,具体执行如下命令:
```python ```python
# 进入unet migraphx工程根目录 # 进入unet migraphx工程根目录
...@@ -109,7 +115,7 @@ cd ./build/ ...@@ -109,7 +115,7 @@ cd ./build/
./Unet ./Unet
``` ```
会在当前目录中生成分割图像 会在当前目录中生成分割图像
<img src="./Doc/Images/Unet_02.jpg" style="zoom:100%;" align=middle> <img src="./Doc/Images/Unet_02.jpg" style="zoom:100%;" align=middle>
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment