DETR tensorRT部署去除推理过程无用辅助头+fp16部署再次加速+解决转tensorrt 输出全为0问题的新方法
特别说明:参考官方开源的DETR代码、TensorRT官方文档,如有侵权告知删,谢谢。 完整代码、测试脚本、测试图片、模型文件点击下载1、转tensorrt输出全为0老问题回顾 在用TensorRT部署DETR检测模型时遇到:转tensorrt输出全为0的问题。多次想放弃这个模型部署,花了很多时间查阅,最终解决方法用了两步: 第一步,修改onnx模型输出层Gather的参数; 第二步,