c++ 目标检测计算交并比

郭建洋

于 2024-03-03 10:17:07 发布

阅读量207

点赞数 3

文章标签： c++ 目标检测算法

本文链接：https://blog.youkuaiyun.com/weixin_46438576/article/details/136427377

版权

本文介绍了如何使用C++实现IOU（IntersectionoverUnion）算法，用于计算两个矩形区域的重叠程度，常用于目标检测中的非极大抑制（NMS）过程。代码展示了如何通过比较交叉点和面积来计算IOU值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

    auto iou = [](const vector<float>& a, const vector<float>& b){
        float cross_left   = std::max(a[0], b[0]);
        float cross_top    = std::max(a[1], b[1]);
        float cross_right  = std::min(a[2], b[2]);
        float cross_bottom = std::min(a[3], b[3]);

        float cross_area = std::max(0.0f, cross_right - cross_left) * std::max(0.0f, cross_bottom - cross_top);
        float union_area = std::max(0.0f, a[2] - a[0]) * std::max(0.0f, a[3] - a[1]) 
                         + std::max(0.0f, b[2] - b[0]) * std::max(0.0f, b[3] - b[1]) - cross_area;
        if(cross_area == 0 || union_area == 0) return 0.0f;
        return cross_area / union_area;
    };

这里是引用
iou相当于一个函数，其输入为a和b。输出为iou的值。
调用时如下：

            if(ibox[4] == jbox[4]){
                // class matched
                if(iou(ibox, jbox) >= nms_threshold)
                    remove_flags[j] = true;
            }