摘要
本申请公开了一种植物病虫害识别模型的构建方法、装置、设备及介质。其中,该方法基于轻量级网络、自注意力机制、特征融合层和分类器,构建初始模型;采用样本数据集对初始模型进行训练得到中间模型;对中间模型进行轻量化处理得到植物病虫害识别模型,以基于植物病虫害识别模型对待识别植物图像进行病虫害识别。本技术方案,通过结合轻量级网络和自注意力机制搭建初始模型,以使模型在对图像特征进行高效提取的同时,提高对图像中长距离依赖信息的捕捉能力;进而对训练好的中间模型进行轻量化处理,以保证模型在移动设备和嵌入式设备上的高效运行,同时减少对计算资源的消耗,从而提高了对植物病虫害的识别精度和识别速度。
技术关键词
植物病虫害
瓶颈
注意力机制
剪枝模型
识别植物
网络
分类器
图像
数据
样本
模型训练模块
可读存储介质
嵌入式设备
计算机
蒸馏
处理器通信
移动设备
存储器