单个图像的地面实况掩蔽
IDtrn包含唯一的文件夹名称每个文件夹有2个子文件夹:
- 图片
- 口罩
此代码适用于我:
for i in IDtrn:
Ytrain = []
temp = glob.glob(os.path.join(BASE_PATH, i, 'masks', '*'))
mask = np.zeros((256, 256))
for j in temp:
mask_ = cv2.imread(j, cv2.IMREAD_GRAYSCALE)
mask_ = cv2.resize(mask_, (256, 256))
mask = np.maximum(mask, mask_)
Ytrain = Ytrain + [mask]
Ytrain = np.asarray(Ytrain)
img = Ytrain[len(temp)-1].astype(np.uint8)
img = Image.fromarray(img)
img.convert('RGB')
img.save(os.path.join(BASE_PATH, i, 'masks', 'combined_mask.png'))