如何调整从Deeplab v3获得的分割蒙版的大小？

Question

Deeplab v3返回缩小/调整大小的图像及其对应的蒙版。如何调整图像的大小以及相应的蒙版，使其更适合我的规格。

Answer 1

cv2.resize方法可用于将插值方法保持为cv2.INTER_NEAREST

resized_image = cv2.resize(segmentation_mask, target_dims, interpolation 
=cv2.INTER_NEAREST)

这种插​​值方法不会导致遮罩中存在的标签的RGB值发生变化。

如果要在调整大小后保存遮罩，请保持格式为“ .png”。其他格式往往会少量改变像素值，这对于分段蒙版是不希望的。