Research on Chinese Medicine Image Recognition Based on an Improved Swin Transformer Model
With the rapid development of deep learning technology,computer vision has shown tremen-dous application prospects in multiple fields.Chinese herbal medicine recognition is of great signifi-cance for ensuring the quality of Chinese herbal medicine.By optimizing the model structure and intro-ducing the Squeeze-and-Excitation attention mechanism,a Chinese herbal medicine image recognition method based on an improved Swin Transformer model was proposed.The experimental results based on the publicly available dataset of Chinese herbal medicine images showed that the improved model significantly improved the recognition accuracy of Chinese herbal medicine images.Meanwhile,the smaller loss function also indicated that the new model had strong generalization ability.
Deep learningChinese herbal medicine recognitionSwin TransformerAttention mechanism
万方数据
