[1]LIU Wanjun,TONG Chang,QU Haicheng.An antagonistic image shadow removal algorithm based on dilated convolution and attention mechanism[J].CAAI Transactions on Intelligent Systems,2021,16(6):1081-1089.[doi:10.11992/tis.202011022]
Copy
CAAI Transactions on Intelligent Systems[ISSN 1673-4785/CN 23-1538/TP] Volume:
16
Number of periods:
2021 6
Page number:
1081-1089
Column:
学术论文—智能系统
Public date:
2021-11-05
- Title:
-
An antagonistic image shadow removal algorithm based on dilated convolution and attention mechanism
- Author(s):
-
LIU Wanjun; TONG Chang; QU Haicheng
-
Software College, Liaoning Technical University, Huludao 125105, China
-
- Keywords:
-
generative adversarial networks; hole convolution; multiple attention; residual network; multi-scale; autocoder; long short-term memory; shadow removal
- CLC:
-
TP391.41
- DOI:
-
10.11992/tis.202011022
- Abstract:
-
To solve the problem of the unobvious shadow removal effect in dark areas or complex textured and penumbra areas, an antagonistic image shadow removal algorithm is proposed based on a dilated convolution and attention mechanism. The algorithm is based on the general idea of generative adversarial networks. First, the dilated convolution is introduced into the residual network, and the user-defined hole residual block is used for feature extraction, expanding the receptive field of feature extraction. Second, in the attention coding stage, four layers of dilated convolution with the same structure are added to provide more abstract and essential global semantic features for the decoding phase with a minimum calculation amount. Finally, the multiple attention mechanism is used to guide the discrimination network to identify the unshadowed image; thus, improving the discrimination network’s ability. The proposed algorithm is tested on image shadow triplets dataset and shadow removal public datasets and achieves the structural similarity of 97.77%. The image feature information of the algorithm is well preserved, the picture is clear, the shadow removal effect is good in the dark area and complex area, and the algorithm has good performance for the penumbra area.