Machine learning based image segmentation training with contour accuracy evaluation

Abstract

Improving the accuracy of predicted segmentation masks, including: extracting a ground-truth RGB image buffer and a binary contour image buffer from a ground-truth RGB image container for segmentation training; generating predicted segmentation masks from the ground-truth RGB image buffer; generating second binary contours from the predicted segmentation masks using a particular algorithm; computing a segmentation loss between manually-segmented masks of the ground-truth RGB image buffer and the predicted segmentation masks; computing a contour accuracy loss between contours of the binary contour image buffer and the binary contours of the predicted segmentation masks; computing a total loss as a weighted average of the segmentation loss and the contour accuracy loss; and generating improved binary contours by compensating the contours of the binary contour image buffer with the computed total loss, wherein the improved binary contours are used to improve the accuracy of the predicted segmentation masks.