Detection and Counting of Small Target Apples under Complicated Environments by Using Improved YOLOv7-tiny

被引:51
作者
Ma, Li [1 ]
Zhao, Liya [1 ]
Wang, Zixuan [1 ]
Zhang, Jian [2 ,3 ]
Chen, Guifen [4 ]
机构
[1] Jilin Agr Univ, Coll Informat Technol, Changchun 130118, Peoples R China
[2] Jilin Agr Univ, Fac Agron, Changchun 130118, Peoples R China
[3] Univ Columbia Okanagan, Dept Biol, Kelowna, BC V1V 1V7, Canada
[4] Changchun Humanities & Sci Coll, Inst Technol, Changchun 130118, Peoples R China
来源
AGRONOMY-BASEL | 2023年 / 13卷 / 05期
基金
中国国家自然科学基金;
关键词
YOLOv7-tiny-Apple; small target; fruit detection and counting; digital agriculture;
D O I
10.3390/agronomy13051419
中图分类号
S3 [农学(农艺学)];
学科分类号
0901 ;
摘要
Weather disturbances, difficult backgrounds, the shading of fruit and foliage, and other elements can significantly affect automated yield estimation and picking in small target apple orchards in natural settings. This study uses the MinneApple public dataset, which is processed to construct a dataset of 829 images with complex weather, including 232 images of fog scenarios and 236 images of rain scenarios, and proposes a lightweight detection algorithm based on the upgraded YOLOv7-tiny. In this study, a backbone network was constructed by adding skip connections to shallow features, using P2BiFPN for multi-scale feature fusion and feature reuse at the neck, and incorporating a lightweight ULSAM attention mechanism to reduce the loss of small target features, focusing on the correct target and discard redundant features, thereby improving detection accuracy. The experimental results demonstrate that the model has an mAP of 80.4% and a loss rate of 0.0316. The mAP is 5.5% higher than the original model, and the model size is reduced by 15.81%, reducing the requirement for equipment; In terms of counts, the MAE and RMSE are 2.737 and 4.220, respectively, which are 5.69% and 8.97% lower than the original model. Because of its improved performance and stronger robustness, this experimental model offers fresh perspectives on hardware deployment and orchard yield estimation.
引用
收藏
页数:17
相关论文
共 44 条
[1]   An Artificial Intelligence-Based Stacked Ensemble Approach for Prediction of Protein Subcellular Localization in Confocal Microscopy Images [J].
Aggarwal, Sonam ;
Gupta, Sheifali ;
Gupta, Deepali ;
Gulzar, Yonis ;
Juneja, Sapna ;
Alwan, Ali A. ;
Nauman, Ali .
SUSTAINABILITY, 2023, 15 (02)
[2]  
Bargoti Suchet, 2017, 2017 IEEE International Conference on Robotics and Automation (ICRA), P3626, DOI 10.1109/ICRA.2017.7989417
[3]   A weakly-supervised approach for flower/fruit counting in apple orchards [J].
Bhattarai, Uddhav ;
Karkee, Manoj .
COMPUTERS IN INDUSTRY, 2022, 138
[4]   The results of consumer preference testing of popular apple cultivars at the end of the storage season [J].
Blazek, Jan ;
Paprstein, Frantisek ;
Zeleny, Lubor ;
Krelinova, Jana .
HORTICULTURAL SCIENCE, 2019, 46 (03) :115-122
[5]   EFFECTIVE FEATURE FUSION NETWORK IN BIFPN FOR SMALL OBJECT DETECTION [J].
Chen, Jun ;
Mai, HongSheng ;
Luo, Linbo ;
Chen, Xiaoqiang ;
Wu, Kangle .
2021 IEEE INTERNATIONAL CONFERENCE ON IMAGE PROCESSING (ICIP), 2021, :699-703
[6]   A Multiscale Lightweight and Efficient Model Based on YOLOv7: Applied to Citrus Orchard [J].
Chen, Junyang ;
Liu, Hui ;
Zhang, Yating ;
Zhang, Daike ;
Ouyang, Hongkun ;
Chen, Xiaoyan .
PLANTS-BASEL, 2022, 11 (23)
[7]  
Chuanjiang Zhang, 2020, 2020 Proceedings of IEEE International Conference on Artificial Intelligence and Computer Applications (ICAICA), P267, DOI 10.1109/ICAICA50127.2020.9181907
[8]   CenterNet: Keypoint Triplets for Object Detection [J].
Duan, Kaiwen ;
Bai, Song ;
Xie, Lingxi ;
Qi, Honggang ;
Huang, Qingming ;
Tian, Qi .
2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, :6568-6577
[9]  
Ge Z, 2021, Arxiv, DOI [arXiv:2107.08430, DOI 10.48550/ARXIV.2107.08430]
[10]   NAS-FPN: Learning Scalable Feature Pyramid Architecture for Object Detection [J].
Ghiasi, Golnaz ;
Lin, Tsung-Yi ;
Le, Quoc V. .
2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, :7029-7038