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

修改readme文档

parent 82baf0df
......@@ -8,34 +8,38 @@ U-net模型整体为U型结构,主要分为三部分:下采样、上采样
## Python版本推理
下面介绍如何运行python代码示例,具体推理代码解析,在Doc/Tutorial_Python.md中有详细说明
下面介绍如何运行python代码示例,Python示例的详细说明见Doc目录下的Tutorial_Python.md。
### 拉取镜像
### 下载镜像
在光源可拉取推理的docker镜像,Unet模型推理的镜像如下
在光源中下载MIGraphX镜像
```python
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
# 进入unet migraphx工程根目录
cd <path_to_unet_migraphx>
# 进入示例程序目录
cd ./Python/
cd Python/
# 安装依赖
pip install -r requirements.txt
```
3.在Python目录下执行如下命令运行该示例程序:
### 运行示例
在Python目录下执行如下命令运行该示例程序:
```python
python Unet.py
......@@ -47,9 +51,15 @@ python Unet.py
## 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依赖
......@@ -60,10 +70,8 @@ sh ./3rdParty/InstallOpenCVDependences.sh
### 修改CMakeLists.txt
- 如果使用ubuntu系统,需要修改CMakeLists.txt中依赖库路径:
将"${CMAKE_CURRENT_SOURCE_DIR}/depend/lib64/"修改为"${CMAKE_CURRENT_SOURCE_DIR}/depend/lib/"
- **MIGraphX2.3.0及以上版本需要c++17**
如果使用ubuntu系统,需要修改CMakeLists.txt中依赖库路径:
将"${CMAKE_CURRENT_SOURCE_DIR}/depend/lib64/"修改为"${CMAKE_CURRENT_SOURCE_DIR}/depend/lib/"
### 安装OpenCV并构建工程
......@@ -94,9 +102,7 @@ export LD_LIBRARY_PATH=<path_to_unet_migraphx>/depend/lib/:$LD_LIBRARY_PATH
source ~/.bashrc
```
### 推理示例
运行Unet示例程序,具体执行如下命令:
### 运行示例
```python
# 进入unet migraphx工程根目录
......@@ -109,7 +115,7 @@ cd ./build/
./Unet
```
会在当前目录中生成分割图像
会在当前目录中生成分割图像
<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