使用numpy计算两个框的iou
需求简单,记录一下。
程序
import numpy as np
import cv2def iou(bbox1, bbox2):
xmin1, ymin1, xmax1, ymax1 = bbox1
xmin2, ymin2, xmax2, ymax2 = bbox2# 获取矩形框交集对应的顶点坐标(intersection)
xx1 = np.max([xmin1, xmin2])
yy1 = np.max([ymin1, ymin2])
xx2 = np.min([xmax1, xmax2])
yy2 = np.min([ymax1, ymax2])
# 计算交集面积
inter_area = (np.max([0, xx2 - xx1])) * (np.max([0, yy2 - yy1]))# 计算两个矩形框面积
area1 = (xmax1 - xmin1 ) * (ymax1 - ymin1)
area2 = (xmax2 - xmin2) * (ymax2 - ymin2)# 计算交并比(交集/并集)
iou = inter_area / (area1 + area2 - inter_area )# 注意:这里inter_area不能乘以2,乘以2就相当于把交集部分挖空了
return ioudef main():
img = np.zeros((600,600), dtype=np.uint8)+255# 创建一个全白画板(灰度图)
img = np.stack((img,)*3, axis=-1)# 将灰度图转化为三通道图。目的是能在画板上画出彩色的图案。# 初始化两个框
bbox1 = [200, 200, 400, 400]# xmin, ymin, xmax, ymax
bbox2 = [275, 300, 325, 550]# xmin, ymin, xmax, ymax# 在画板上画出这两个彩色框
cv2.rectangle(img, (bbox1[0:2]), (bbox1[2:4]), (255,69,0), -1)
cv2.rectangle(img, (bbox2[0:2]), (bbox2[2:4]), (218,112,214), -1)# 计算这两个框的iou
rst = iou(bbox1, bbox2)cv2.putText(img, 'iou: %.6f' % rst, (10,30), cv2.FONT_HERSHEY_SIMPLEX, 1, (0,0,255), 2)
cv2.imwrite('res.png',img)if __name__ == "__main__":
main()
结果
【使用numpy计算两个框的iou】
文章图片
推荐阅读
- 由浅入深理解AOP
- 【译】20个更有效地使用谷歌搜索的技巧
- mybatisplus如何在xml的连表查询中使用queryWrapper
- MybatisPlus|MybatisPlus LambdaQueryWrapper使用int默认值的坑及解决
- MybatisPlus使用queryWrapper如何实现复杂查询
- iOS中的Block
- Linux下面如何查看tomcat已经使用多少线程
- 使用composer自动加载类文件
- android|android studio中ndk的使用
- 使用协程爬取网页,计算网页数据大小