CPC H04N 7/147 (2013.01) [G06F 18/214 (2023.01); G06F 18/24765 (2023.01); G06N 3/04 (2013.01); G06N 3/08 (2013.01); G06T 7/11 (2017.01); G06T 7/194 (2017.01); G06V 10/82 (2022.01); G06V 30/19173 (2022.01); G06V 30/242 (2022.01); G06T 2207/10016 (2013.01); G06T 2207/10024 (2013.01); G06T 2207/20024 (2013.01); G06T 2207/20081 (2013.01); G06T 2207/20084 (2013.01); G06T 2207/20221 (2013.01); G06T 2207/30201 (2013.01); H04N 5/44504 (2013.01); H04N 5/76 (2013.01); H04N 7/141 (2013.01)] | 19 Claims |
1. A method comprising:
generating, by a processor of a device, an image of a user of the device;
generating an initial mask by applying a segmentation neural network to the image of the user;
refining borders of labeled areas of the initial mask by applying a post-processing engine to the initial mask to generate an output mask;
generating a modified image based on the output mask; and
storing the modified image on the device,
wherein the post-processing engine comprises a guided filter using a downsampled image of the image and a grayscale image of the downsampled image to generate the output mask,
wherein the guided filter further uses an eroded mask of the initial mask and a resized mask of the eroded mask to generate the output mask.
|