针对yolov5网络输出的结果，对数据做后处理以输出目标检测结果
engine网络后处理
包含：
（1）传入一张图片转为需要的格式
（2）调用engine进行推理了
（3）对输出的后处理
（4）输出结果绘图
需要注意的是：
1 pytorch的pt文件转.onnx文件的时候涉及batchsize值，onnx2engine的时候也需要设置batchsize值，infer推理的时候也有batchsize参数。上述三处的batchsize的值需要一致，否则会出现【Cuda failure: 700 已放弃 (核心已转储)】的错误提示。
2 代码仅演示了一张固定图片的推理操作。
代码：
1 CMakeLists.txt
2 postprocess.cpp
若有描述或理解错误，请指正，非常感谢！
相关代码链接：链接：https://pan.baidu.com/s/1IUPyVb15PpWpnJ4lrLVFNQ
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容，请联系我们，一经查实，本站将立刻删除。
如需转载请保留出处：https://51itzy.com/kjqy/203981.html