Research on adaptive object detection via improved HSA‐YOLOv5 for raspberry maturity detection
Abstract In the field of machine vision, target detection models have experienced rapid development and have been practically applied in various domains. In agriculture, target detection models are commonly used to identify various types of fruits. However, when it comes to recognizing berries, such...
Saved in:
| Main Authors: | , , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
Wiley
2024-12-01
|
| Series: | IET Image Processing |
| Subjects: | |
| Online Access: | https://doi.org/10.1049/ipr2.13149 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1846121455439314944 |
|---|---|
| author | Chen Ling Qunying Zhang Mei Zhang Chihan Gao |
| author_facet | Chen Ling Qunying Zhang Mei Zhang Chihan Gao |
| author_sort | Chen Ling |
| collection | DOAJ |
| description | Abstract In the field of machine vision, target detection models have experienced rapid development and have been practically applied in various domains. In agriculture, target detection models are commonly used to identify various types of fruits. However, when it comes to recognizing berries, such as raspberries, the fruits nearing ripeness exhibit highly similar colours, posing a challenge for existing target detection models to accurately identify raspberries in this stage. Addressing this issue, a raspberry detection method called HSA‐YOLOv5 (HSV self‐adaption YOLOv5) is proposed. This method detects immature, nearly ripe, and ripe raspberries. The approach involves transforming the RGB colour space of the original dataset images into an improved HSV colour space. By adjusting corresponding parameters and enhancing the contrast of similar colours while retaining the maximum features of the original image, the method strengthens data features. Adaptive selection of HSV parameters is performed based on data captured under different weather conditions, applying homogeneous preprocessing to the dataset. The improved model is compared with the original YOLOv5 model using a self‐constructed dataset. Experimental results demonstrate that the improved model achieves a mean average precision (mAP) of 0.97, a 6.42 percentage point increase compared to the baseline YOLOv5 model. In terms of immature, nearly ripe, and ripe raspberries, there are improvements of 6, 4, and 7 percentage points, respectively, validating the effectiveness of the proposed model. |
| format | Article |
| id | doaj-art-cb324ed3ce0d4bc89f950af6f90ac30f |
| institution | Kabale University |
| issn | 1751-9659 1751-9667 |
| language | English |
| publishDate | 2024-12-01 |
| publisher | Wiley |
| record_format | Article |
| series | IET Image Processing |
| spelling | doaj-art-cb324ed3ce0d4bc89f950af6f90ac30f2024-12-16T04:00:31ZengWileyIET Image Processing1751-96591751-96672024-12-0118144898491210.1049/ipr2.13149Research on adaptive object detection via improved HSA‐YOLOv5 for raspberry maturity detectionChen Ling0Qunying Zhang1Mei Zhang2Chihan Gao3The Electrical Engineering College Guizhou University Guiyang ChinaGuizhou Botanical Garden Guiyang ChinaThe Electrical Engineering College Guizhou University Guiyang ChinaThe Electrical Engineering College Guizhou University Guiyang ChinaAbstract In the field of machine vision, target detection models have experienced rapid development and have been practically applied in various domains. In agriculture, target detection models are commonly used to identify various types of fruits. However, when it comes to recognizing berries, such as raspberries, the fruits nearing ripeness exhibit highly similar colours, posing a challenge for existing target detection models to accurately identify raspberries in this stage. Addressing this issue, a raspberry detection method called HSA‐YOLOv5 (HSV self‐adaption YOLOv5) is proposed. This method detects immature, nearly ripe, and ripe raspberries. The approach involves transforming the RGB colour space of the original dataset images into an improved HSV colour space. By adjusting corresponding parameters and enhancing the contrast of similar colours while retaining the maximum features of the original image, the method strengthens data features. Adaptive selection of HSV parameters is performed based on data captured under different weather conditions, applying homogeneous preprocessing to the dataset. The improved model is compared with the original YOLOv5 model using a self‐constructed dataset. Experimental results demonstrate that the improved model achieves a mean average precision (mAP) of 0.97, a 6.42 percentage point increase compared to the baseline YOLOv5 model. In terms of immature, nearly ripe, and ripe raspberries, there are improvements of 6, 4, and 7 percentage points, respectively, validating the effectiveness of the proposed model.https://doi.org/10.1049/ipr2.13149agricultural engineeringcomputer visionimage processing |
| spellingShingle | Chen Ling Qunying Zhang Mei Zhang Chihan Gao Research on adaptive object detection via improved HSA‐YOLOv5 for raspberry maturity detection IET Image Processing agricultural engineering computer vision image processing |
| title | Research on adaptive object detection via improved HSA‐YOLOv5 for raspberry maturity detection |
| title_full | Research on adaptive object detection via improved HSA‐YOLOv5 for raspberry maturity detection |
| title_fullStr | Research on adaptive object detection via improved HSA‐YOLOv5 for raspberry maturity detection |
| title_full_unstemmed | Research on adaptive object detection via improved HSA‐YOLOv5 for raspberry maturity detection |
| title_short | Research on adaptive object detection via improved HSA‐YOLOv5 for raspberry maturity detection |
| title_sort | research on adaptive object detection via improved hsa yolov5 for raspberry maturity detection |
| topic | agricultural engineering computer vision image processing |
| url | https://doi.org/10.1049/ipr2.13149 |
| work_keys_str_mv | AT chenling researchonadaptiveobjectdetectionviaimprovedhsayolov5forraspberrymaturitydetection AT qunyingzhang researchonadaptiveobjectdetectionviaimprovedhsayolov5forraspberrymaturitydetection AT meizhang researchonadaptiveobjectdetectionviaimprovedhsayolov5forraspberrymaturitydetection AT chihangao researchonadaptiveobjectdetectionviaimprovedhsayolov5forraspberrymaturitydetection |