python - 提取检测到的近似形状和边界框Mask RCNN

Question

RCNN Mask 检测人添加边界框和形状近似，我的目标是只获得这些
如何仅从 RCNN Mask 获得形状近似和边界框？

Answer 1

假设您有一个带有对象掩码的 NumPy 数组列表，您可以尝试 scikit-image 的 regionprops函数，它返回相应边界框的列表(以及其他有用的属性):

from skimage import measure
a = np.array([[0,0,0],[0,1,0],[0,1,0],[0,1,0],[0,0,0]])
b = measure.regionprops(a)
print(b[0].bbox)

返回: (1, 1, 4, 2) ，对应于 (min_row, min_col, max_row, max_col) .请注意，根据 https://scikit-image.org/docs/dev/api/skimage.measure.html#regionprops ，边界框坐标的间隔是半开的(前包含，后排除)如下: [min_row; max_row)和 [min_col; max_col) .