python - 将边界框区域转换为 mask 并将其另存为PNG文件

Question

我有一组图像及其边界框值，如下所示:

image_names class   xmin    xmax    ymin    ymax
image1.png  1       260     361     45      184

我希望将这些边界框值转换为具有边界框尺寸并用白色像素填充的蒙版，并以“image1_mask.png”格式存储每个图像的蒙版。我们有预定义的功能来做到这一点吗？

Answer 1

简单的方法

import cv2
import numpy as np

img = cv2.imread('image1.png') # read image
mask = np.zeros((img.shape[0],img.shape[1]),dtype=np.uint8) # initialize mask
mask[ymin:ymax,xmin:xmax] = 255 # fill with white pixels
cv2.imwrite('image1_mask.png',mask) # save mask