CPC G06T 7/12 (2017.01) [G06F 18/214 (2023.01); G06V 10/44 (2022.01); G06V 10/774 (2022.01); G06T 2207/10024 (2013.01)] | 15 Claims |
1. A method for improving the accuracy of predicted segmentation masks, the method comprising:
extracting a ground-truth red-green-blue (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 second 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.
|