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