基于改进CenterNet的柑橘叶片病害检测

Disease detection of citrus leaves based on improved CenterNet

针对柑橘叶片病害表现大小不一,为解决检测过程中出现的漏检、误检、准确率不高的问题,提出了改进CenterNet模型。在特征提取网络RestNet50的前两个残差层的一系列残差结构中引入特征增强改进的空洞卷积池化金字塔(improved atrous spatial pyramid pooling,IASPP)模块,扩大浅层感受野,获取更多小目标叶片病害的细节信息,增强浅层特征的显著性;引入双向加权特征融合模块(bidirectional feature pyramid network,BiFPN),有效融合特征提取网络浅层和深层叶片病害信息;为提高整体检测效果,引入多尺度通道注意力机制(multi-scale channel attention module,MS-CAM)。训练后的模型对柑橘病害叶片进行检测,实验结果表明,相比于原模型CenterNet,所提模型的R值提高了8.32%,mAP提高了4.53%,AP_(0.5∶0.95)上升了27.3%,可实现柑橘种植中对叶片小目标、中目标、大目标病害的精准检测。

Aiming at the different sizes of the disease manifestations of citrus leaves,the model of improved CenterNet is proposed to solve the problems of missing detection,false detection and low accuracy in the detection process.The feature enhancement improved atrous spatial pyramid pooling(IASPP)module was introduced into a series of residual structures of the first two residual layers of the feature extraction network RestNet50 to expand the shallow receptive field,obtain more detailed information of small target leaf diseases,and enhance the significance of shallow features.The bidirectional feature pyramid network(BiFPN)module was introduced to effectively integrate the shallow and deep leaf disease information of the feature extraction network.In order to improve the overall detection effect,the multi-scale channel attention module(MS-CAM)was introduced.The trained model was used to detect citrus disease leaves.The experimental results show that,compared with the original model CenterNet,theR-value of the proposed model is increased by 8.32%,mAPis increased by 4.53%,andAP_(0.5:0.95)is increased by 27.3%.It can achieve accurate detection of small target,medium target and large target leaf diseases in citrus planting.