General Electric Company
MULTI-STAGE SEGMENTATION USING SYNTHETIC IMAGES

Abstract:

The example embodiments are directed to refinement process for generating an accurate image segmentation map. A refinement network may enhance an initially generated segmentation map using a model that is trained using synthetic images. In one example, the method may include storing an image of content which includes a plurality of categories of data, receiving an initial segmentation map of the image, the initial segmentation map comprising pixel probability values with respect to the plurality of categories, executing a refinement predictive model on the initial segmentation map and the image to generate a refined segmentation map, wherein the predictive model is trained using synthetic images of the plurality of categories of data, and generating a segmented image based on the refined segmentation map.