haotian/duan8
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update 使用TRT加速网络-零.md

Browse Source
This commit is contained in:
BaofengZan 2020-09-21 17:16:41 +08:00 committed by wang-xinyu
parent 3c2c644d27
commit ad93465ab5
1 changed files with 3 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
HRNetClassification/使用TRT加速网络-零.md
View File

@ -82,14 +82,8 @@ ResBlock层
搭建网络时需要从wts中查看层名，各个卷积层信息需要从代码中分析。
其中卷积层后面的size信息为：
$$
InKsize * OutKsize * Kszie*Ksize
$$
比如第二个卷积层： 卷积层 的输入维度64 输出维度64 卷积核大小3*3
$$
64*64*3*3=36864
$$
![image_f](https://user-images.githubusercontent.com/20653176/93750398-fd341c00-fc2d-11ea-9077-ee749b6aef41.png)
![image-20200918142959711](https://user-images.githubusercontent.com/20653176/93749484-8fd3bb80-fc2c-11ea-951d-3c1f403e521a.png)
## 5 TRT搭建网络
@ -134,4 +128,4 @@ network->markOutput(*out->getOutput(0));
## 7 TRT代码整理
这里就是将TRT搭建的网络，能封装函数，就封装为函数模块，增加代码可读性。
这里就是将TRT搭建的网络，能封装函数，就封装为函数模块，增加代码可读性。