河南农业科学 ›› 2022, Vol. 51 ›› Issue (11): 155-162.DOI: 10.15933/j.cnki.1004-3268.2022.11.018
王川1,赵恒滨1,2,李国强2,3,张建涛2,3,高桐梅4,赵巧丽2,3,郑国清2,3
WANG Chuan1,ZHAO Hengbin1,2,LI Guoqiang2,3,ZHANG Jiantao2,3,GAO Tongmei4,ZHAO Qiaoli2,3,ZHENG Guoqing2,3
摘要: 为实现密集条件下芝麻蒴果的准确检测,提出基于YOLOX模型的芝麻蒴果检测定位方法(CEYOLOX模型)。该模型以CSPDarknet-53作为主干特征提取网络,在路径聚合网络PANet中增加104×104大尺度特征层,增强对目标细粒度特征信息的获取;通过引入注意力机制模块获取目标重要的轮廓特征和空间位置信息;将传统的NMS替换为更有利于重叠目标检测的Soft-NMS算法来降低漏检情况。结果表明,在IoU阈值为0.5时,CE-YOLOX模型在全部测试集上的调和均值(F1)、召回率、平均精度分别为0.99、98.65%、99.71%,与原模型YOLOX相比,该模型分别提升了0.05、6.27个百分点、3.28个百分点。通过蒴果计数试验,CE-YOLOX模型计数准确率为96.84%,比YOLOX模型提高了5.28个百分点。改进后的模型CE-YOLOX适用于密集条件下芝麻蒴果检测。
中图分类号: