Как изменить размер маски сегментации, полученной из Deeplab v3?

Deeplab v3 возвращает уменьшенное/измененное в размере изображение и соответствующую ему маску. Как я могу изменить размер изображения, а также его соответствующую маску, чтобы она лучше соответствовала моей спецификации.


person Ahmad Bilal Khalid    schedule 24.12.2018    source источник


Ответы (1)


Можно использовать метод cv2.resize, сохраняя метод интерполяции cv2.INTER_NEAREST.

resized_image = cv2.resize(segmentation_mask, target_dims, interpolation 
=cv2.INTER_NEAREST)

Этот метод интерполяции не приведет к изменению значений RGB меток, присутствующих в маске.

Если вы сохраняете маски после изменения размера, сохраните формат «.png». Другие форматы имеют тенденцию изменять значения пикселей на небольшую величину, что нежелательно для масок сегментации.

person Rachna Pathak    schedule 24.04.2019