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

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()