• • 下一篇
刘馨瑶1,梁军1,龙嘉濠1,颜仁梁2
Fine-grained Chinese herbal medicine image classification based on feature fusion and channel information compensation
摘要: 在传统中草药细粒度图像分类领域,缺乏一个全面且平衡的数据集。为推进中草药细粒度图像识别研究,构建了Herb-150细粒度中草药数据集。针对该任务中深层神经网络易丢失判别性细节特征的问题,提出了细粒度特征增强的CHMRN网络,通过引入自顶向下的特征融合模块整合多尺度语义信息,从而捕捉全面的上下文特征;同时,设计了自底向上的通道信息补偿模块,以增强细粒度特征的表达能力,确保能够准确捕捉中药类别之间的细微差异。实验结果表明,CHMRN网络在Herb-150数据集上的准确率达到了93.91%,相比细粒度图像分类模型CMAL-Net、PIM提升了超过3%,同时优于IELT、SR-GNN、I2-HOFI、SIM-OFE模型,验证了所提架构在细粒度分类任务中的有效性。CHMRN网络不仅提高了传统中药识别的准确性,还能为类似的细粒度图像分类应用提供参考。
中图分类号: