Comment redimensionner le masque de segmentation obtenu à partir de Deeplab v3 ?

Question

Deeplab v3 renvoie une image réduite/redimensionnée et son masque correspondant. Comment puis-je redimensionner l'image ainsi que le masque correspondant pour qu'ils correspondent mieux à mes spécifications ?

Answer 1

La méthode cv2.resize peut être utilisée en conservant la méthode d'interpolation cv2.INTER_NEAREST.

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

Cette méthode d'interpolation n'entraînera pas de changement dans les valeurs RVB des étiquettes présentes dans le masque.

Si vous enregistrez les masques après les avoir redimensionnés, conservez le format '.png'. Les autres formats ont tendance à modifier les valeurs des pixels de façon minime, ce qui n'est pas souhaitable pour les masques de segmentation.