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
UNet_migraphx
Commits
5d519488
Commit
5d519488
authored
Jun 13, 2023
by
liucong
Browse files
修改readme文档
parent
82baf0df
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
24 additions
and
18 deletions
+24
-18
README.md
README.md
+24
-18
No files found.
README.md
View file @
5d519488
...
@@ -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
>
...
...
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