- 不分类别,无差别nms
- 按照类别nms
f32.data是numpy保存的二进制数据:
(1)读取f32.data保存到std::vector
(2)保留第四列数值大于conf_thres的行;(xywhc+类);
(3)找最大的类别分类数值概率,然后乘以第四列数值,然后只有大于conf_thres的行才会被保留;同时转换数据到x1y1x2y2score;
(4)升序排序;
(5)nms,保留iou符合要求的。
#include#include #include #include using namespace std; float iou



