首页|基于改进ThreshNet模型的农作物虫害识别方法

基于改进ThreshNet模型的农作物虫害识别方法

扫码查看
针对现有深度学习模型在复杂背景下虫害特征提取能力差、泛化性能低等问题,提出一种基于改进脱粒网模型的农作物虫害识别方法.引入批量通道归一化模块,提高模型的泛化能力;将高层筛选特征金字塔网络与自定义卷积模块融合,形成多尺度特征融合模块,把模块嵌入密集连接的模型与谐波之间,增强模型特征提取能力;调整模型整体架构,得到驱虫脱粒网(TNP)模型.通过自建数据集P28 对比实验,结果表明,与改进前模型相比,TNP模型的准确率提高了3.95%、参数量下降了 6.47M、浮点运算量(FLOPs)下降了 0.23G;与 ResNet50、DenseNet、EfficientNet B4 等模型相比,TNP模型的准确率、参数量、FLOPs和推理时间均有较好表现.TNP模型能够快速、准确识别农作物虫害特征信息,为及时防治虫害提供技术支持.
Crop Pest Identification Method Based on Improved ThreshNet Model
Aiming at the problems of poor feature extraction ability and low generalization performance of existing deep learning models for pests in complex backgrounds,a crop pest recognition method based on improved ThreshNet model is proposed.Firstly,a bulk channel normalization module(BCN)is introduced to improve the generalization ability of the model;secondly,a high level screening feature pyramid network(HSFPN)is fused with a customized convolution module(partial-conv)to form a multi-scale feature fusion module(PCHS).Secondly,the module is embedded into the model between the dense connections and the harmonic dense connections to enhance the model's feature extraction capability.Finally,the overall architecture of the model is adjusted to obtain the improved TNP model.Comparison experiments are carried out on the self-built dataset P28.The experimental results show that compared with the pre-improvement model,the accuracy of the TNP model increases by 3.95%,the number of parameters decreases by 6.47M,and the amount of floating-point operations(FLOPs)decreases by 0.23G.Compared withResNet50,DenseNet,EfficientNet B4 and other models,the TNP model has better performance in terms of accuracy,parameter quantity,FLOPs and inference time.The improved model can quickly and accurately identify the characteristic information of crop pests,and provide technical support for timely prevention and control of pests.

cropspest identificationThreshNet modelBCN modulemulti-scale feature fusion

任喜伟、孙悦、杨虹、何立风

展开 >

陕西科技大学电子信息与人工智能学院,西安 710021

西安富德医疗电子有限公司,西安 710075

农作物 虫害识别 ThreshNet模型 BCN模块 多尺度特征融合

2024

实验室研究与探索
上海交通大学

实验室研究与探索

CSTPCD北大核心
影响因子:1.69
ISSN:1006-7167
年,卷(期):2024.43(12)