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...

Descripción completa

Detalles Bibliográficos
Autores principales: Liu, Pingzhu, Yin, Hua
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