Improve segmentation to increase yield