BA Auto-correction of Segmentation Masks