Ground/Flood masks postprocessing

[melisandeteng]

There are two main things for which postprocessing steps could be used in ground/flood masks obtained with Deeplab models:

• Blobs in the sky / Small noisy disconnected components:

Ideas for criteria

• Look at surrounding pixels of a connected component - e.g. reject if all the pixels around are segmented as sky
• Look at location of the blob if the image: reject if > x % of the blob if in the top y % part of the image
  • Holes in the ground:
• the close_holes.py script closes all holes that are smaller than a chosen threshold. What is a good threshold ? How can we adapt it depending on the input image?
  If it is too big, the postprocessing step might lead to even more erroneous masks. Example below: ground segmentation with Deeplabv3 and ground segmentation with the postprocessing step.

[Shivamshaiv]

The method of dealing with such problems, in general, would be to use a combination of the distance transform of the binary marks and followed by morphological erosion and dilation. The best function implementation to do these tasks would be

1. scipy.ndimage.morphology.distance_transform_edt
2. skimage.morphology.remove_small_objects