Image restoration method and device, and non-transitory computer storage medium
AssigneeBoe Technology Group Co., Ltd.
InventorMengdi SUN
Life of this patent
Estimating the term…
Abstract
Disclosed are an image restoration method and device, and a non-transitory computer storage medium. The image restoration method includes: an original mask in an original to-be-restored image is processed into a dilated mask with a regular shape, the dilated mask is overlaid at the location of the original mask in the original to-be-restored image to acquire a mask enhancement feature map, and then the dilated mask in the mask enhancement feature map is restored using an image restoration network to acquire a restored image corresponding to the original to-be-restored image.
Claims21
An image restoration method, comprising:
- acquiring an original to-be-restored image;
- performing mask detection on the original to-be-restored image;
- acquiring an original mask feature map of the original to-be-restored image in response to detecting an original mask in the original to-be-restored image;
- acquiring a dilated mask feature map by processing the original mask in the original mask feature map into a dilated mask, wherein the dilated mask comprises at least one mask unit, and the original mask is located in an area in which the at least one mask unit is located;
- acquiring a mask enhancement feature map by overlaying the dilated mask in the dilated mask feature map at a location of the original mask in the original to-be-restored image;
- and
- acquiring a restored image corresponding to the original to-be-restored image by inputting the mask enhancement feature map into an image restoration network to restore the dilated mask in the mask enhancement feature map.
-15. (canceled)
An image restoration device, comprising a processor and a memory storing at least one instruction, at least one program, a code set, or an instruction set thereon, wherein the processor, when loading and executing the at least one instruction, the at least one program, the code set, or the instruction set, is caused to perform:
- acquiring an original to-be-restored image;
- performing mask detection on the original to-be-restored image;
- acquiring an original mask feature map of the original to-be-restored image in response to detecting an original mask in the original to-be-restored image;
- acquiring a dilated mask feature map by processing the original mask in the original mask feature map into a dilated mask, wherein the dilated mask comprises at least one mask unit, and the original mask is located in an area in which the at least one mask unit is located;
- acquiring a mask enhancement feature map by overlaying the dilated mask in the dilated mask feature map at a location of the original mask in the original to-be-restored image;
- and
- acquiring a restored image corresponding to the original to-be-restored image by inputting the mask enhancement feature map into an image restoration network to restore the dilated mask in the mask enhancement feature map.
A non-transitory computer storage medium, storing at least one instruction, at least one program, a code set, or an instruction set thereon, wherein the at least one instruction, the at least one program, the code set, or the instruction set, when loaded and executed by a processor, causes the processor to perform:
- acquiring an original to-be-restored image;
- performing mask detection on the original to-be-restored image;
- acquiring an original mask feature map of the original to-be-restored image in response to detecting an original mask in the original to-be-restored image;
- acquiring a dilated mask feature map by processing the original mask in the original mask feature map into a dilated mask, wherein the dilated mask comprises at least one mask unit, and the original mask is located in an area in which the at least one mask unit is located;
- acquiring a mask enhancement feature map by overlaying the dilated mask in the dilated mask feature map at a location of the original mask in the original to-be-restored image;
- and
- acquiring a restored image corresponding to the original to-be-restored image by inputting the mask enhancement feature map into an image restoration network to restore the dilated mask in the mask enhancement feature map.
Dependencies are inferred from the claim text.