科研之友
微信
新浪微博
Facebook
分享链接
摘要
本发明公开了一种基于注意力机制的金字塔网络的中草药识别方法,包括以下步骤：1)构建中草药数据集,制作中草药训练集与中草药测试集；2)构建基于通道注意力机制的特征融合结构块,引入竞争注意力模块；3)在金字塔网络的特征融合结构块添加空间注意力机制,使用空间协同注意力模块,对两条信息流进行调整,将调整后的两条信息流融合作为输出；4)构造基于注意力机制的金字塔网络,用中草药训练集进行训练；5)将中草药测试集中的图片传入训练好的网络模型,识别出该图片对应的中草药类别。本发明提高了中草药识别的准确率与性能,并且除了能辅助相关产业人员识别中草药,也能方便非专业人士对中草药的辨别。
- 单位
