Chinese traditional painting style automatic classification based on dual-channel feature fusion with multi-attention mechanism

Abstract

Existing classification models for traditional Chinese paintings mostly ignore shallow detail features, which leads to the imprecise classification of styles. To address the above problems, this paper proposes a Chinese traditional painting style automatic classification model based on dual-channel feature fusion with multi-attention mechanism. First, the spatial attention mechanism is introduced to enhance the Swin-Transformer framework to obtain the salient features of Chinese ancient painting images. Second, a dual-channel attention mechanism is constructed to extract global semantic features and local features of Chinese ancient painting images. Finally, the extracted features are fused and categorized based on the softmax classifier. To verify the feasibility and validity of the proposed model, this paper performs simulations on the Chinese painting dataset and compares it with existing algorithms.The average classification accuracy of the proposed model is 90.6[Formula: see text], with an improvement of 3.14[Formula: see text], which is better than the existing model in both visual effects and objective data comparisons.