Update segmentation.yaml