一种融合注意力机制的轻量级重楼饮片分类方法

A LIGHTWEIGHT PARIS L. DECOCTION PIECES RECOGNITION METHOD BASED ON ATTENTION MECHANISM

  • 摘要: 针对重楼属植物中毛重楼、滇重楼、白花重楼的自动化识别问题,提出一种融合注意力机制的轻量级重楼饮片分类模型。首先,提出两种多尺度特征提取模块,综合提取多种尺度特征。然后,在ECA-Net (Efficient Channel Attention Network)和空间注意力机制的基础上提出ECSA-Module (Efficient Channel and Spatial Attention Module),使特征图通道和空间信息得到充分利用。最后对主干网络进行密集连接,并使用随机擦除方法进行数据增强。实验表明,该模型的分类准确率高达96.74%,相较于MobileNet-V2、VGG16、Xception等模型分别提高了3.26百分点、2.82百分点、2.22百分点。基于该模型开发出的重楼识别系统精度高、速度快,具有重要的实践应用价值。

     

    Abstract: Aimed at the automatic recognition of Paris mairei H. Lév., Paris polyphylla var. yunnanensis (Franch.) Hand.-Mzt. and Paris polyphylla Sm. Var. Alba H Li et R. J., a lightweight Paris L. decoction pieces classification model based on attention mechanism is proposed. Two multi-scale feature extraction modules were proposed to comprehensively extract multiple scale features. On the basis of ECA-Net and spatial attention mechanism, ECSA-Module (Efficient channel and spatial attention module) was proposed to make full use of feature map channels and spatial information. The backbone network was densely connected, and the random erasing method was used for data enhancement. The experimental results show that the classification accuracy of the model is as high as 96.74%, which is 3.26 percentage points, 2.82 percentage points and 2.22 percentage points higher than that of MobileNet-V2, VGG16 and Xception respectively. The recognition system of Paris L. based on this model has high recognition accuracy and high speed, which has important practical application value.

     

/

返回文章
返回