diff --git a/HRNetClassification/使用TRT加速网络-零.md b/HRNetClassification/使用TRT加速网络-零.md index 06a05ff..51565b7 100644 --- a/HRNetClassification/使用TRT加速网络-零.md +++ b/HRNetClassification/使用TRT加速网络-零.md @@ -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搭建的网络,能封装函数,就封装为函数模块,增加代码可读性。 \ No newline at end of file +这里就是将TRT搭建的网络,能封装函数,就封装为函数模块,增加代码可读性。