Fixed a problem in segmentation that would leave pixels with value 1, thus falsifying the geometry features