如何调整从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”。其他格式倾向于将像素值改变少量，这对于分割掩模是不期望的。