将图像分割的 mask 转换成目标检测的 bbox

最新推荐文章于 2025-03-25 21:42:44 发布

原创最新推荐文章于 2025-03-25 21:42:44 发布 · 9.1k 阅读

24 ·

CC 4.0 BY-SA版权

我们不生产知识，我们只是互联网的搬运工

图像处理与计算机视觉专栏收录该内容

33 篇文章

订阅专栏

本文介绍了一种基于连通组件分析的目标检测方法，通过OpenCV库实现图像中多个目标的定位与绘制边界框，展示了如何使用Python进行图像处理和目标识别。

在这里插入图片描述

def find_bbox(mask):
    _, labels, stats, centroids = cv2.connectedComponentsWithStats(mask.astype(np.uint8))
    stats = stats[stats[:,4].argsort()]
    return stats[:-1]

mask = label[4]
ax = plt.axes()
plt.imshow(mask,cmap='bone')
bboxs = find_bbox(mask)
for j in bboxs: 
    rect = patches.Rectangle((j[0],j[1]),j[2],j[3],linewidth=1,edgecolor='r',facecolor='none')
    ax.add_patch(rect)
plt.show()