Cargando…
YOLOv7-Peach: An Algorithm for Immature Small Yellow Peaches Detection in Complex Natural Environments
Using object detection techniques on immature fruits to find out their quantity and position is a crucial step for intelligent orchard management. A yellow peach target detection model (YOLOv7-Peach) based on the improved YOLOv7 was proposed to address the problem of immature yellow peach fruits in...
Autores principales: | , |
---|---|
Formato: | Online Artículo Texto |
Lenguaje: | English |
Publicado: |
MDPI
2023
|
Materias: | |
Acceso en línea: | https://www.ncbi.nlm.nih.gov/pmc/articles/PMC10255365/ https://www.ncbi.nlm.nih.gov/pubmed/37299824 http://dx.doi.org/10.3390/s23115096 |
_version_ | 1785056853976154112 |
---|---|
author | Liu, Pingzhu Yin, Hua |
author_facet | Liu, Pingzhu Yin, Hua |
author_sort | Liu, Pingzhu |
collection | PubMed |
description | Using object detection techniques on immature fruits to find out their quantity and position is a crucial step for intelligent orchard management. A yellow peach target detection model (YOLOv7-Peach) based on the improved YOLOv7 was proposed to address the problem of immature yellow peach fruits in natural scenes that are similar in color to the leaves but have small sizes and are easily obscured, leading to low detection accuracy. First, the anchor frame information from the original YOLOv7 model was updated by the K-means clustering algorithm in order to generate anchor frame sizes and proportions suitable for the yellow peach dataset; second, the CA (coordinate attention) module was embedded into the backbone network of YOLOv7 so as to enhance the network’s feature extraction for yellow peaches and to improve the detection accuracy; then, we accelerated the regression convergence process of the prediction box by replacing the object detection regression loss function with EIoU. Finally, the head structure of YOLOv7 added the P2 module for shallow downsampling, and the P5 module for deep downsampling was removed, effectively improving the detection of small targets. Experiments showed that the YOLOv7-Peach model had a 3.5% improvement in mAp (mean average precision) over the original one, much higher than that of SSD, Objectbox, and other target detection models in the YOLO series, and achieved better results under different weather conditions and a detection speed of up to 21 fps, suitable for real-time detection of yellow peaches. This method could provide technical support for yield estimation in the intelligent management of yellow peach orchards and also provide ideas for the real-time and accurate detection of small fruits with near background colors. |
format | Online Article Text |
id | pubmed-10255365 |
institution | National Center for Biotechnology Information |
language | English |
publishDate | 2023 |
publisher | MDPI |
record_format | MEDLINE/PubMed |
spelling | pubmed-102553652023-06-10 YOLOv7-Peach: An Algorithm for Immature Small Yellow Peaches Detection in Complex Natural Environments Liu, Pingzhu Yin, Hua Sensors (Basel) Article Using object detection techniques on immature fruits to find out their quantity and position is a crucial step for intelligent orchard management. A yellow peach target detection model (YOLOv7-Peach) based on the improved YOLOv7 was proposed to address the problem of immature yellow peach fruits in natural scenes that are similar in color to the leaves but have small sizes and are easily obscured, leading to low detection accuracy. First, the anchor frame information from the original YOLOv7 model was updated by the K-means clustering algorithm in order to generate anchor frame sizes and proportions suitable for the yellow peach dataset; second, the CA (coordinate attention) module was embedded into the backbone network of YOLOv7 so as to enhance the network’s feature extraction for yellow peaches and to improve the detection accuracy; then, we accelerated the regression convergence process of the prediction box by replacing the object detection regression loss function with EIoU. Finally, the head structure of YOLOv7 added the P2 module for shallow downsampling, and the P5 module for deep downsampling was removed, effectively improving the detection of small targets. Experiments showed that the YOLOv7-Peach model had a 3.5% improvement in mAp (mean average precision) over the original one, much higher than that of SSD, Objectbox, and other target detection models in the YOLO series, and achieved better results under different weather conditions and a detection speed of up to 21 fps, suitable for real-time detection of yellow peaches. This method could provide technical support for yield estimation in the intelligent management of yellow peach orchards and also provide ideas for the real-time and accurate detection of small fruits with near background colors. MDPI 2023-05-26 /pmc/articles/PMC10255365/ /pubmed/37299824 http://dx.doi.org/10.3390/s23115096 Text en © 2023 by the authors. https://creativecommons.org/licenses/by/4.0/Licensee MDPI, Basel, Switzerland. This article is an open access article distributed under the terms and conditions of the Creative Commons Attribution (CC BY) license (https://creativecommons.org/licenses/by/4.0/). |
spellingShingle | Article Liu, Pingzhu Yin, Hua YOLOv7-Peach: An Algorithm for Immature Small Yellow Peaches Detection in Complex Natural Environments |
title | YOLOv7-Peach: An Algorithm for Immature Small Yellow Peaches Detection in Complex Natural Environments |
title_full | YOLOv7-Peach: An Algorithm for Immature Small Yellow Peaches Detection in Complex Natural Environments |
title_fullStr | YOLOv7-Peach: An Algorithm for Immature Small Yellow Peaches Detection in Complex Natural Environments |
title_full_unstemmed | YOLOv7-Peach: An Algorithm for Immature Small Yellow Peaches Detection in Complex Natural Environments |
title_short | YOLOv7-Peach: An Algorithm for Immature Small Yellow Peaches Detection in Complex Natural Environments |
title_sort | yolov7-peach: an algorithm for immature small yellow peaches detection in complex natural environments |
topic | Article |
url | https://www.ncbi.nlm.nih.gov/pmc/articles/PMC10255365/ https://www.ncbi.nlm.nih.gov/pubmed/37299824 http://dx.doi.org/10.3390/s23115096 |
work_keys_str_mv | AT liupingzhu yolov7peachanalgorithmforimmaturesmallyellowpeachesdetectionincomplexnaturalenvironments AT yinhua yolov7peachanalgorithmforimmaturesmallyellowpeachesdetectionincomplexnaturalenvironments |