Jetson Nano 【14】Pytorch的YOLOv3 spp 模型中关于tensorRT+ 矩形推理问题的解决过程

背景说明

• 模型是YOLOv3 spp
• 框架是Pytorch
• 由于我的Nano上GPU计算资源不够，所以我们急需减少模型计算量，在剪枝与量化+转tensorRT后从500ms达到了85ms每张，但依然达不到要求，于是想到了矩形推理。

最终效果

• 在256x416的视频中实现50-55ms 处理一张图片的成绩，换算成FPS 在 20左右（YOLOv3 Spp）。

实现过程

• 还在整理，过段时间会上传至我的git，欢迎关注。