wq_0708的博客

以YOLOv5s为例，使用TensorRT Python API搭建yolov5网络

使用Python API实现TRT版BatchNorm算子BatchNorm定义TRT的Scale层定义Scale实现BNTensorRT未提供批量归一化层BatchNorm，但提供了更通用的Scale层。可以使用Scale层来实现BN层。BatchNorm定义BN[i,:]=in[i,:]−mean[i]var[i]+ϵ∗γ[i]+β[i]BN[i,:]=\frac{in[i,:]-mean[i]}{\sqrt{var[i]+\epsilon}}*\gamma[i]+\beta[i]BN[i,:

C++语言实现TensorRT7自定义插件——主要包含两大类“插件类”和“插件创建器类”

使用Python API构建TensorRT engine

使用C++ API构建TensorRT engine

本文以TensorRT7.2.3.4为例讲解Ubuntu在Docker中安装TensorRT