2023年5月重庆师范大学学报(自然科学版)May2023第40卷第3期JournalofChongqingNormalUniversity(NaturalScience)Vol.40No.3DOI:10.11721/cqnuj20230305融合大核注意力机制的土壤种类识别网络*刘康1,2,曾绍华1,2,胡静3,刘国一4,李超5(1.重庆师范大学计算机与信息科学学院,重庆401331;2.重庆市数字农业服务工程技术研究中心,重庆401331;3.重庆市农村土地整治中心,重庆401121;4.西藏自治区农牧科学院农业资源与环境研究所,拉萨850032;5.重庆市铜梁区农业农村委员会,重庆铜梁402560)摘要:【目的】为解决因土壤图像纹理复杂、没有结构性特征导致的传统卷积神经网络模型难以提取其中关键性特征、识别准确率低的问题,提出了一种大核注意力(largekernelattention,LKA)机制模块与ResNet模型融合的土壤种类识别网络模型VAR(visualattentionResNet),以解决土壤种类样本不平衡和难分类样本造成的模型泛化能力弱的问题。【方法】以ResNet为主干网络,在主干中引入LKA机制,减少其中的残差块,构建土壤种类识别网络VAR,并改进网络的焦点损失函数(FocalLoss)。【结果】1)与传统模型ResNet18、ResNet34、VGG、GooleNet、VAN等相比,VAR模型在特定模型参数下对紫色土土壤图像数据集中土壤种类的识别精度更高;2)用3种不同大小VAR模型之一的VAR_small与以ResNet18为主干并嵌入传统注意力机制SE、CBAM、ECA和SK的网络进行对比,实验结果显示LKA机制在土壤识别方面更加优秀;3)改进的FocalLoss可让VAR更能注意到难分类的土壤图像样本。【结论】将LKA机制模块与ResNet模型融合的土壤种类识别网络模型VAR增强了网络提取土壤图像中关键性结构特征能力,同时还减少了网络参数,能更加有效地识别土壤种类。关键词:土壤种类识别;VAR;大核注意力;焦点损失函数中图分类号:TP391.41文献标志码:A文章编号:1672-6693(2023)03-0129-10土壤是农作物生产的基础,农作物的种类选择、种植方式、田间管理等均受到土壤种类差异的影响,因而正确识别土壤种类对农业...