Как изменить размер маски сегментации, полученной из 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».Другие форматы имеют тенденцию изменять значения пикселей на небольшую величину, что нежелательно для масок сегментации.