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
ResNet50_onnxruntime
Commits
91faedd2
Commit
91faedd2
authored
Oct 11, 2023
by
yangql
Browse files
修改格式
parent
3b0f65fc
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
10 additions
and
8 deletions
+10
-8
README.md
README.md
+10
-8
No files found.
README.md
View file @
91faedd2
...
...
@@ -7,12 +7,12 @@
## 模型结构
ResNet50模型包含了49个卷积层、一个全连接层。


## 算法原理
ResNet50使用了多个具有残差连接的残差块来解决梯度消失或梯度爆炸问题,并使得网络可以向更深层发展。


## 环境配置
### Docker
...
...
@@ -23,7 +23,7 @@ docker pull image.sourcefind.cn:5000/dcu/admin/base/custom:ort1.14.0_migraphx3.0
创建并启动容器:
```
docker run --shm-size 16g --network=host --name=resnet50_ort --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v $PWD/resnet50_o
rt
:/home/resnet50_o
rt
-it <Your Image ID> /bin/bash
docker run --shm-size 16g --network=host --name=resnet50_ort --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v $PWD/resnet50_o
nnxruntime
:/home/resnet50_o
nnxruntime
-it <Your Image ID> /bin/bash
# 激活dtk
source /opt/dtk/env.sh
...
...
@@ -41,8 +41,8 @@ export PYTHONPATH=/opt/dtk/lib:$PYTHONPATH
```
#### 运行示例
```
python
# 进入resnet50 o
rt
工程根目录
cd
<
path_to_resnet50_o
rt
>
# 进入resnet50 o
nnxruntime
工程根目录
cd
<
path_to_resnet50_o
nnxruntime
>
# 进入示例程序目录
cd
Python
/
...
...
@@ -79,13 +79,15 @@ export LD_LIBRARY_PATH=<path_to_resnet50_onnxruntime>/depend/lib64/:$LD_LIBRARY_
```
source ~/.bashrc
```
####下载模型
sh ./Resource/Models/download.sh
#### 运行示例
```
# 进入resnet50
ort
工程根目录
# 进入resnet50
_onnxruntime
工程根目录
cd <path_to_resnet50_onnxruntime>
#下载模型
sh ./Resource/Models/download.sh
# 进入build目录
cd build/
...
...
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