Generating refined segmentation masks based on uncertain pixels

A system comprising:

• at least one memory device comprising a segmentation refinement neural network;
• and
• at least one server device configured to cause the system to:
• receive a digital visual media item comprising a plurality of pixels and depicting one or more objects;
• and
• utilize the segmentation refinement neural network to:
• generate a set of feature maps corresponding to the digital visual media item comprising a set of initial feature maps and a final feature map;
• generate an initial segmentation mask by determining whether the plurality of pixels correspond to the one or more objects based on the final feature map;
• determine, based on the initial segmentation mask, uncertain pixels, the uncertain pixels having an associated uncertainty that the uncertain pixels correspond to the one or more objects or do not correspond to the one or more objects;
• extract, from the set of initial feature maps, feature values associated with the uncertain pixels;
• and
• generate a refined segmentation mask for the digital visual media item by redetermining whether a set of uncertain pixels correspond to the one or more objects based on the feature values associated with the uncertain pixels.

In a digital medium environment for editing digital visual media, a computer-implemented method for generating segmentation masks comprising:

• receiving, from a computing device, a digital visual media item comprising a plurality of pixels, the digital visual media item depicting one or more objects;
• and
• utilizing a segmentation refinement neural network to:
• generate an initial segmentation mask corresponding to the digital visual media item by determining whether the plurality of pixels correspond to the one or more objects by generating probabilities that the plurality of pixels correspond to the one or more objects;
• determine, based on the initial segmentation mask, uncertain pixels, the uncertain pixels having an associated uncertainty that the uncertain pixels correspond to the one or more objects or do not correspond to the one or more objects;
• and
• generate a refined segmentation mask for the digital visual media item by redetermining whether a set of uncertain pixels correspond to the one or more objects by generating updated probabilities that the plurality of pixels correspond to the one or more objects.