如何调整从 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”。其他格式往往会少量改变像素值，这对于分割掩码来说是不可取的。