基于改进的Yolov7的西红柿果实计数算法研究
Tomato Fruit Counting Algorithm Based on Improved Yolov7作者机构:内蒙古农业大学机电工程学院呼和浩特010018 内蒙古第一机械集团股份有限公司包头014010
出 版 物:《内蒙古农业大学学报(自然科学版)》 (Journal of Inner Mongolia Agricultural University(Natural Science Edition))
年 卷 期:2024年第45卷第1期
页 面:48-56页
学科分类:0832[工学-食品科学与工程(可授工学、农学学位)] 08[工学] 083203[工学-农产品加工及贮藏工程]
基 金:内蒙古自治区科技创新引导项目(Kcj1-202205)
摘 要:针对目前统计西红柿果实数效率低、人工成本高、计数算法精确性差的问题,本文提出了一种改进的Yolov7的西红柿果实计数算法,采用CBAM模块对特征值进行自适应加权、强化西红柿特征并降低背景干扰,同时融合BiFPN结构以优化不同尺度特征的权重分配,高效提升特征融合效率,最终达到快速准确的识别效果。试验研究结果显示,该算法在平均计数精度(MAPE)、平均绝对误差(MAE)和均方误差(MSE)等3个关键性能指标上均显著优于其他算法,改进后的Yolov7的MAE、MSE和MAPE分别达到1.63、1.98和5.31%,相比于Yolov4、Yolov5s以及原版Yolov7,MAE和MSE分别降低了3.24和3.15、2.35和2.29以及1.13和1.05,计数误差率比Yolov4、Yolov5s和Yolov7分别减少了3.34%、1.71%和1.53%。