基于卷积块注意力胶囊网络的小样本水稻害虫识别
2022-03-15分类号:TP183;TP391.41;S435.112
【部门】安徽大学农业生态大数据分析与应用技术国家地方联合工程研究中心 安徽大学互联网学院 科大国创软件股份有限公司
【摘要】针对真实复杂背景下小样本水稻害虫识别模型泛化能力弱,易受复杂背景干扰以及重要特征表达能力不强等问题,提出了一种基于卷积块注意力胶囊网络的小样本水稻害虫识别方法。采用数据增强的方法扩充数据,以提高模型的泛化能力同时预防过拟合;利用GrabCut算法去除图像中的复杂背景,减小复杂背景对水稻害虫识别的干扰;将空间注意机制和通道注意机制相结合的卷积块注意模块(Convolutional block attention module,CBAM)引入到胶囊网络中,提高模型对水稻害虫特征的表达能力,使模型关注重要特征,抑制不必要的特征。其中胶囊网络主要用来更加敏锐地发现小样本图像中水稻害虫的相对位置和角度等信息。结果表明:在对复杂背景下小样本水稻害虫的识别时,本研究方法准确识别率达99.19%,优于支持向量机(Support vector machine,SVM)、k近邻(k-nearest neighbors,kNN)等浅层网络方法,也优于VGG16、GoogLeNet以及Mobilenet等深度网络方法,可实现复杂背景下的小样本水稻害虫的准确识别。
【关键词】水稻害虫 害虫识别 卷积神经网络 胶囊网络 注意力机制
【基金】安徽省自然科学基金面上项目(2108085MC95);; 安徽省科技重大专项(202003a06020016);; 安徽省高校自然科学研究项目(KJ2020ZD03,KJ2020A0039);; 农业生态大数据分析与应用技术国家地方联合工程研究中心开放课题(AE202004)
【所属期刊栏目】中国农业大学学报
文献传递