这个屌丝很懒，什么也没留下！

Yolov5后处理代码 | cpu部分

作者：我家自动化 | 2024-03-20 18:32:36

使用核函数对Yolov5推理的结果进行解码并恢复成框
yolov5的输出tensor(n x 85)，其中85是cx, cy, idth, height, objness, classification*80

CPU解码重点
1.避免多余计算，尤其是数学运算（IOU）
2.nms实现优化，例如remove flag并且预先分配内存，reserve对输出分配内存

GPU解码重点
1.表示输出大学不确定的数据，

#include <cuda_runtime.h>
#include <opencv2/opencv.hpp>
#include <stdio.h>
#1
2
3

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/我家自动化/article/detail/275726