pangxing6491的博客

int8

第一步、配置tensorrt环境，具体参见以下链接：https://zongxp.blog.csdn.net/article/details/86077553第二步、pytorch模型转onnx:注意：需要事先训练得到pth模型，也要知道模型的输入输出，比如此次我用的是resnet１８，我设置的输入是（１，３，３２，３２），即batch_size=1,通道数为３，尺寸为３２＊３２，简而言之，就是一次输入一张３２＊３２的彩色ＲＧＢ图像。import torchfrom resnet imp

https://blog.csdn.net/irving512/article/details/114019280

https://zongxp.blog.csdn.net/article/details/86077553