Sketch recognition model based on improved CycleGAN network and dual attention mechanism
摘要
To improve sketch recognition accuracy, this study proposes an enhanced sketch recognition model based on an improved CycleGAN network and a dual attention mechanism. The proposed model first incorporates multi-directional convolution and brightness equalization modules into the CycleGAN network to extract edge and contour features. A dual attention mechanism is then implemented using channel attention and spatial attention modules, effectively addressing issues of sparse strokes and uneven spatial distribution in sketches while enhancing the representation of critical features. Finally, a hybrid architecture combining global average pooling and convolution layers serves as the classifier to produce sketch recognition results. Simulation results demonstrate that this model achieves 97.08% accuracy, 98.12% precision, 98.23% recall, and 97.45% F1 score for sketch recognition on the TU-Berlin dataset, and 98.65% accuracy, 98.12% precision, 98.76% recall, and 97.95% F1 score on the QuickDraw dataset. Compared with state-of-the-art sketch recognition models, this model exhibits superior performance in accuracy. These results indicate that the model can enhance sketch recognition precision and provide technical support for converting sketches into high-quality animated images.