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