The diverse and complex morphological characteristics of traditional Chinese medicine herbs make automated image classification of traditional Chinese medicine a critical and challenging task. To address the issue of limited traditional Chinese medicine image datasets, this study constructs a dataset consisting of 16 common traditional Chinese medicine herb categories and a total of 3552 images, covering a diverse range of backgrounds, freshness levels, processing methods, and display arrangements. Based on the characteristics of the dataset, a specialized feature extraction encoder, CSA (ConvNeXtv2 with SE and ACmix), is proposed for traditional Chinese medicine image classification. The CSA encoder employs large-kernel convolutions to extract high-dimensional feature maps from traditional Chinese medicine images. It then enhances the competition among channels in the feature maps using a squeeze-and-excitation (SE) module and improves the network’s ability to process both global and local information through a convolution-attention mixed module. To reduce computational costs, the output dimension of the ACmix module is reduced to one-quarter of the original channel number, followed by a 1 × 1 convolutional layer to increase the channel dimension to z. Experimental results show that the CSA encoder outperforms current mainstream deep learning algorithms, with improvements of 1.56%, 1.81%, 1.69%, and 2.15% in accuracy, precision, recall, and F1 score, respectively. Ablation studies demonstrate that the integration of both the SE and ACmix modules achieves the best classification performance, validating the effectiveness of these modules.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

A Chinese Medicinal Herb Image Classification Algorithm Combining Local and Global Features

  • Wenbin Dai,
  • Yihao Zhao,
  • Yuduo Chen,
  • Jun Ma

摘要

The diverse and complex morphological characteristics of traditional Chinese medicine herbs make automated image classification of traditional Chinese medicine a critical and challenging task. To address the issue of limited traditional Chinese medicine image datasets, this study constructs a dataset consisting of 16 common traditional Chinese medicine herb categories and a total of 3552 images, covering a diverse range of backgrounds, freshness levels, processing methods, and display arrangements. Based on the characteristics of the dataset, a specialized feature extraction encoder, CSA (ConvNeXtv2 with SE and ACmix), is proposed for traditional Chinese medicine image classification. The CSA encoder employs large-kernel convolutions to extract high-dimensional feature maps from traditional Chinese medicine images. It then enhances the competition among channels in the feature maps using a squeeze-and-excitation (SE) module and improves the network’s ability to process both global and local information through a convolution-attention mixed module. To reduce computational costs, the output dimension of the ACmix module is reduced to one-quarter of the original channel number, followed by a 1 × 1 convolutional layer to increase the channel dimension to z. Experimental results show that the CSA encoder outperforms current mainstream deep learning algorithms, with improvements of 1.56%, 1.81%, 1.69%, and 2.15% in accuracy, precision, recall, and F1 score, respectively. Ablation studies demonstrate that the integration of both the SE and ACmix modules achieves the best classification performance, validating the effectiveness of these modules.