Extreme Image Classification Algorithm Based on Multicore Dense Connection Network

• Main Authors: Daolei Wang, Tianyu Zhang, Rui Zhu, Mingshan Li, Jiajun Sun
• Format: Article
• Language: English
• Published: Hindawi Limited 2021-01-01
• Series: Mathematical Problems in Engineering
• Online Access: http://dx.doi.org/10.1155/2021/6616325

Extreme images refer to low-quality images taken under extreme environmental conditions such as haze, heavy rain, strong light, and shaking, which will lead to the failure of the visual system to effectively recognize the target. Most of the existing extreme image restoration algorithms only handle the restoration work of a certain type of image; how to effectively recognize all kinds of extreme images is still a challenge. Therefore, this paper proposes a classification and restoration algorithm for extreme images. Due to the large differences in the features on extreme images, it is difficult for the existing models such as DenseNet to effectively extract depth features. In order to solve the classification problem in the algorithm, we propose a Multicore Dense Connection Network (MDCNet). MDCNet is composed of dense part, attention part, and classification part. Dense Part uses two dense blocks with different convolution kernel sizes to extract features of different sizes; attention part uses channel attention mechanism and spatial attention mechanism to amplify the effective information in the feature map; classification part is mainly composed of two convolutional layers and two fully connected layers to extract and classify feature images. Experiments have shown that the recall of MDCNet can reach 92.75% on extreme image dataset. At the same time, the mAP value of target detection can be improved by about 16% after the image is processed by the classification and recovery algorithm.