基于TensorRT完成NanoDet模

北京哪家荨麻疹医院好 http://disease.39.net/bjzkbdfyy/220324/10490515.html

为大家介绍一个TensorRTint8量化部署NanoDet模型的教程,并开源了全部代码。主要包括如何搭建tensorrt环境,对pytorch模型做onnx格式转换,onnx模型做tensorrtint8量化,及对量化后的模型做推理,实测在显卡做到了2ms一帧!

NanoDet简介

NanoDet(是一个速度超快和轻量级的Anchor-free目标检测模型;和yolov4tiny作比较(如下图),精度相当,但速度却快了1倍;对于速度优先的场景,nanodet无疑是一个好的选择。




转载请注明:http://www.aierlanlan.com/rzdk/2617.html

  • 上一篇文章:
  •   
  • 下一篇文章: 没有了