面试题-手撕IOU计算

zhaoyqcsdn

已于 2024-01-09 20:55:56 修改

阅读量601

点赞数 7

CC 4.0 BY-SA版权

分类专栏：面试文章标签：自动驾驶 c++ 笔记经验分享面试

于 2024-01-04 20:00:35 首次发布

本文链接：https://blog.youkuaiyun.com/zhaoyqcsdn/article/details/135394772

一般在面试中只会考察二维无旋转矩形框IOU的计算，更复杂的二维框带旋转、三维框等一般不会让写。

二维无旋转矩形框IOU计算

最简单的iou计算，矩形框，无旋转
2个框的排布可能有下面三种情况，都需要考虑到
在这里插入图片描述

代码

#include <iostream>
#include <algorithm>

struct Rectangle {
    float x1, y1, x2, y2; // 左上角和右下角坐标
};

float intersectionArea(const Rectangle& rect1, const Rectangle& rect2) {
    float overlapWidth = std::max(0.0f, std::min(rect1.x2, rect2.x2) - std::max(rect1.x1, rect2.x1));
    float overlapHeight = std::max(0.0f, std::min(rect1.y2, rect2.y2) - std::max(rect1.y1, rect2.y1));

    return overlapWidth * overlapHeight;
}

float calculateIOU(const Rectangle& rect1, const Rectangle& rect2) {
    float area1 = (rect1.x2 - rect1.x1) * (rect1.y2 - rect1.y1);
    float area2 = (rect2.x2 - rect2.x1) * (rect2.y2 - rect2.y1);

    float intersection = intersectionArea(rect