数据处理与应用

基于改进的YOLOv3农作物目标检测算法

展开
  • 1.中国气象局·河南省农业气象保障与应用技术重点开放实验室,郑州 450003
    2.河南中原光电测控技术有限公司,郑州 450047
郭蓓,E-mail:1727581698@qq.com
郭蓓,E-mail:1727581698@qq.com

收稿日期: 2023-10-25

  录用日期: 2023-12-10

  网络出版日期: 2024-04-08

基金资助

中国气象局·河南省农业气象保障与应用技术重点开放实验室研究基金(AMF202203)

Improved YOLOv3 Crop Target Detection Algorithm

Expand
  • 1. CMA·Henan Key Open Laboratory of Agrometeorological Support and Application Technology, Zhengzhou 450003, China
    2. Henan Zhongyuan Photolectric Meassurement and Control Technology Co.,LTD, Zhengzhou 450047, China

Received date: 2023-10-25

  Accepted date: 2023-12-10

  Online published: 2024-04-08

摘要

农作物图像在进行目标检测时,由于作物种植较密集、成像质量不佳等原因严重影响目标检测算法的检测精度。针对存在的问题,提出一种基于YOLOv3的改进算法优化在农作物目标检测的检测性能:对YOLOv3的主干特征提取网络进行优化,利用原网络中输出的4倍降采样特征图对目标进行检测,并且在算法原网络残差块的基础上增加残差单元,以检测目标较小的农作物位置信息;提出高斯衰减函数,对图像中高度重叠的农作物候选框的衰减较强,在有效抑制冗余框的同时也可以有效地降低漏检率;对回归损失函数进行优化改进,用CIOU Loss作为损失函数,使得目标检测过程中最终的目标定位更加精确。将改进的 YOLOv3算法和原 YOLOv3 算法、Faster R-CNN 算法在实拍的玉米作物图像数据集上进行对比实验,结果表明改进后的YOLOv3算法能有效检测农作物小目标,算法检测的平均准确率均值和检测速度都有明显的提升。

本文引用格式

郭蓓, 王贝贝, 张志红, 吴苏, 李鹏, 胡莉婷 . 基于改进的YOLOv3农作物目标检测算法[J]. 农业大数据学报, 2024 , 6(1) : 40 -47 . DOI: 10.19788/j.issn.2096-6369.000006

Abstract

When detecting targets in crop images, the detection accuracy of target detection algorithms can be seriously affected due to factors such as dense crop planting and poor imaging quality. In order to optimize the detection performance of crop object detection in YOLOv3, an improved algorithm based on YOLOv3 is proposed. Firstly, the backbone feature extraction network of YOLOv3 is optimized by utilizing the downsampling feature maps outputted by the original network to detect targets, and residual units are added on the basis of the residual blocks in the original network to detect the position information of small crop objects. Moreover, a Gaussian decay function is introduced to attenuate highly overlapping crop candidate boxes in the image, effectively suppressing redundant boxes and reducing false negative rate. Furthermore, the regression loss function is optimized by using CIOU Loss, making the final object localization more accurate during the object detection process. To evaluate the improved YOLOv3 algorithm, a comparative experiment is conducted on a real-world dataset of maize crop images, comparing it with the original YOLOv3 algorithm and the Faster R-CNN algorithm. The results demonstrate that the improved YOLOv3 algorithm can effectively detect small crop targets, exhibiting significantly improved mean average precision and detection speed.

参考文献

[1] 耿创, 宋品德, 曹立佳. YOLO算法在目标检测中的研究进展[J]. 兵器装备工程学报, 2022, 43(9):162-173.
[2] 李维刚, 杨潮, 蒋林, 等. 基于改进YOLOv4算法的室内场景目标检测[J]. 激光与光电子学进展, 2022, 59(18):251-260.
[3] 于博文, 吕明. 改进的YOLOv3算法及其在军事目标检测中的应用[J]. 兵工学报, 2022, 43(2):345-354.
[4] 陈禹蒲, 马晓川, 李璇. 基于YOLOv3锚框优化的侧扫声呐图像目标检测[J]. 信号处理, 2022, 38(11):2359-2371.
[5] 王潇, 李子琦, 高涛, 等. 基于无损跨尺度特征融合的交通目标检测算法[J]. 中国公路学报, 2023, 36(9):315-325.
[6] 王阳, 袁国武, 瞿睿, 等. 基于改进YOLOv3的机场停机坪目标检测方法[J]. 郑州大学学报(理学版), 2022, 54(5):22-28.
[7] 申志超. 基于YOLO的颗粒状农作物检测算法研究[D]. 哈尔滨: 哈尔滨工业大学, 2022.
[8] 张颖超. 基于深度学习的农作物叶片病害检测识别方法及其应用研究[D]. 石家庄: 河北科技大学, 2022.
[9] 金沙沙. 图像分割与目标检测在农业场景中的应用研究[D]. 浙江湖州: 湖州师范学院, 2022.
[10] 贾世娜. 基于改进YOLOv5的小目标检测算法研究[D]. 南昌: 南昌大学, 2022.
[11] 潘昕晖, 邵清, 卢军国. 基于CBD-YOLOv3的小目标检测算法[J]. 小型微型计算机系统, 2022, 43(10):2143-2149.
[12] 鞠默然, 罗海波, 王仲博, 等. 改进的YOLO V3算法及其在小目标检测中的应用[J]. 光学学报, 2019, 39(7):253-260.
[13] 阮激扬. 基于YOLO的目标检测算法设计与实现[D]. 北京: 北京邮电大学, 2019.
[14] 潘语豪, 危疆树, 曾令鹏. 基于YOLOv3的农田鸟类目标检测算法[J]. 激光与光电子学进展, 2022, 59(2):510-518.
[15] 王杨, 曹铁勇, 杨吉斌, 等. 基于YOLO v5算法的迷彩伪装目标检测技术研究[J]. 计算机科学, 2021, 48(10):226-232.
[16] 王佩. 基于特征融合与信息增强的目标检测算法研究[D]. 江西赣州: 江西理工大学, 2022.
[17] 王英立, 史肖波. 一种基于多尺度yolo算法的车辆目标识别方法:CN202210806937.2[P].CN202210806937.2[2023-07-06].
[18] 孙景兰, 张志红, 余卫东, 等. 中国农业气象观测自动化技术研究进展[J]. 气象科技进展, 2022, 12 (4): 7-13.
[19] 王少博, 张成, 苏迪, 等. 基于改进YOLOv3和核相关滤波算法的旋转弹目标探测算法[J]. 兵工学报, 2022, 43(5):1032-1045.
[20] 杨富强, 余波, 赵嘉彬, 等. 基于改进YOLOv3的桥梁底部裂缝目标检测方法[J]. 中国科技论文, 2022, 17(3):252-259.
[21] Lin T Y, Goyal P, Girshick R, et al. Focal loss for dense object detection[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2017, PP(99):2999-3007. https://doi.org/10.48550/arXiv.1708.02002.
[22] Law H, Deng J. Cornernet: Detecting objects as paired keypoints[C]// European Conference on Computer Vision. Springer, Cham, 2018. https://doi.org/10.1007/s11263-019-01204-1
[23] Tian Z, Shen C, Chen H, et al. FCOS: Fully convolutional one-stage object detection[C]// CVF International Conference on Computer Vision (ICCV). IEEE, 2019. https://doi.org/10.1109/ICCV.2019.00972.
[24] Cai Z, Vasconcelos N. Cascade R-CNN: Delving into high quality object detection[C]// Computer Vision and Pattern Recognition (cs.CV), IEEE, 2017. https://doi.org/10.48550/arXiv.1712.00726.
[25] Pang J, Chen K, Shi J, et al. Libra R-CNN: Towards balanced learning for object detection[C]// CVF Conference on Computer Vision and Pattern Recognition (CVPR). IEEE, 2020. https://doi.org/10.48550/arXiv.1904.02701.
[26] Lu X, Li B, Yue Y, et al. Grid R-CNN[C]// CVF Conference on Computer Vision and Pattern Recognition (CVPR). IEEE, 2019. DOI: 10.1109/CVPR.2019.00754.
[27] Redmon J, Divvala S, Girshick R, et al. You Only Look Once: Unified, Real-Time Object Detection[C]// Conference on Computer Vision and Pattern Recognition (CVPR), IEEE, 2019. https://doi.org/10.1109/CVPR.2016.91.
[28] Redmon J, Farhadi A. YOLO9000:Better, Faster, Stronger[C]// Conference on Computer Vision and Pattern Recognition (CVPR). IEEE, 2017:6517-6525. https://doi.org/10.1109/CVPR.2017.690.
文章导航

/