Yolov5s-CA: An Improved Yolov5 Based on the Attention Mechanism for Mummy Berry Disease Detection
Early detection and accurately rating the level of plant diseases plays an important role in protecting crop quality and yield. The traditional method of mummy berry disease (causal agent: <i>Monilinia vaccinii-corymbosi</i>) identification is mainly based on field surveys by crop protec...
| Main Authors: | , , , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
MDPI AG
2022-12-01
|
| Series: | Agriculture |
| Subjects: | |
| Online Access: | https://www.mdpi.com/2077-0472/13/1/78 |
| _version_ | 1797447313224892416 |
|---|---|
| author | Efrem Yohannes Obsie Hongchun Qu Yong-Jiang Zhang Seanna Annis Francis Drummond |
| author_facet | Efrem Yohannes Obsie Hongchun Qu Yong-Jiang Zhang Seanna Annis Francis Drummond |
| author_sort | Efrem Yohannes Obsie |
| collection | DOAJ |
| description | Early detection and accurately rating the level of plant diseases plays an important role in protecting crop quality and yield. The traditional method of mummy berry disease (causal agent: <i>Monilinia vaccinii-corymbosi</i>) identification is mainly based on field surveys by crop protection experts and experienced blueberry growers. Deep learning models could be a more effective approach, but their performance is highly dependent on the volume and quality of labeled data used for training so that the variance in visual symptoms can be incorporated into a model. However, the available dataset for mummy berry disease detection does not contain enough images collected and labeled from a real-field environment essential for making highly accurate models. Complex visual characteristics of lesions due to overlapping and occlusion of plant parts also pose a big challenge to the accurate estimation of disease severity. This may become a bigger issue when spatial variation is introduced by using sampling images derived from different angles and distances. In this paper, we first present the “cut-and-paste” method for synthetically augmenting the available dataset by generating additional annotated training images. Then, a deep learning-based object recognition model Yolov5s-CA was used, which integrates the Coordinated Attention (CA) module on the Yolov5s backbone to effectively discriminate useful features by capturing channel and location information. Finally, the loss function GIoU_loss was replaced by CIoU_loss to improve the bounding box regression and localization performance of the network model. The original Yolov5s and the improved Yolov5s-CA network models were trained on real, synthetic, and combined mixed datasets. The experimental results not only showed that the performance of Yolov5s-CA network model trained on a mixed dataset outperforms the baseline model trained with only real field images, but also demonstrated that the improved model can solve the practical problem of diseased plant part detection in various spatial scales with possible overlapping and occlusion by an overall precision of 96.30%. Therefore, our model is a useful tool for the estimation of mummy berry disease severity in a real field environment. |
| first_indexed | 2024-03-09T13:54:08Z |
| format | Article |
| id | doaj.art-803252094ffd4a4488414215c74710d3 |
| institution | Directory Open Access Journal |
| issn | 2077-0472 |
| language | English |
| last_indexed | 2024-03-09T13:54:08Z |
| publishDate | 2022-12-01 |
| publisher | MDPI AG |
| record_format | Article |
| series | Agriculture |
| spelling | doaj.art-803252094ffd4a4488414215c74710d32023-11-30T20:45:24ZengMDPI AGAgriculture2077-04722022-12-011317810.3390/agriculture13010078Yolov5s-CA: An Improved Yolov5 Based on the Attention Mechanism for Mummy Berry Disease DetectionEfrem Yohannes Obsie0Hongchun Qu1Yong-Jiang Zhang2Seanna Annis3Francis Drummond4College of Computer Science, Chongqing University of Posts and Telecommunications, Chongqing 400065, ChinaCollege of Automation, Chongqing University of Posts and Telecommunications, Chongqing 400065, ChinaSchool of Biology and Ecology, University of Maine, Orono, ME 04469, USASchool of Biology and Ecology, University of Maine, Orono, ME 04469, USASchool of Biology and Ecology, University of Maine, Orono, ME 04469, USAEarly detection and accurately rating the level of plant diseases plays an important role in protecting crop quality and yield. The traditional method of mummy berry disease (causal agent: <i>Monilinia vaccinii-corymbosi</i>) identification is mainly based on field surveys by crop protection experts and experienced blueberry growers. Deep learning models could be a more effective approach, but their performance is highly dependent on the volume and quality of labeled data used for training so that the variance in visual symptoms can be incorporated into a model. However, the available dataset for mummy berry disease detection does not contain enough images collected and labeled from a real-field environment essential for making highly accurate models. Complex visual characteristics of lesions due to overlapping and occlusion of plant parts also pose a big challenge to the accurate estimation of disease severity. This may become a bigger issue when spatial variation is introduced by using sampling images derived from different angles and distances. In this paper, we first present the “cut-and-paste” method for synthetically augmenting the available dataset by generating additional annotated training images. Then, a deep learning-based object recognition model Yolov5s-CA was used, which integrates the Coordinated Attention (CA) module on the Yolov5s backbone to effectively discriminate useful features by capturing channel and location information. Finally, the loss function GIoU_loss was replaced by CIoU_loss to improve the bounding box regression and localization performance of the network model. The original Yolov5s and the improved Yolov5s-CA network models were trained on real, synthetic, and combined mixed datasets. The experimental results not only showed that the performance of Yolov5s-CA network model trained on a mixed dataset outperforms the baseline model trained with only real field images, but also demonstrated that the improved model can solve the practical problem of diseased plant part detection in various spatial scales with possible overlapping and occlusion by an overall precision of 96.30%. Therefore, our model is a useful tool for the estimation of mummy berry disease severity in a real field environment.https://www.mdpi.com/2077-0472/13/1/78wild blueberry<i>Vaccinium angustifolium</i><i>Monilinia vaccinii-corymbosi</i>deep learningcoordinated attentionsynthetic data |
| spellingShingle | Efrem Yohannes Obsie Hongchun Qu Yong-Jiang Zhang Seanna Annis Francis Drummond Yolov5s-CA: An Improved Yolov5 Based on the Attention Mechanism for Mummy Berry Disease Detection Agriculture wild blueberry <i>Vaccinium angustifolium</i> <i>Monilinia vaccinii-corymbosi</i> deep learning coordinated attention synthetic data |
| title | Yolov5s-CA: An Improved Yolov5 Based on the Attention Mechanism for Mummy Berry Disease Detection |
| title_full | Yolov5s-CA: An Improved Yolov5 Based on the Attention Mechanism for Mummy Berry Disease Detection |
| title_fullStr | Yolov5s-CA: An Improved Yolov5 Based on the Attention Mechanism for Mummy Berry Disease Detection |
| title_full_unstemmed | Yolov5s-CA: An Improved Yolov5 Based on the Attention Mechanism for Mummy Berry Disease Detection |
| title_short | Yolov5s-CA: An Improved Yolov5 Based on the Attention Mechanism for Mummy Berry Disease Detection |
| title_sort | yolov5s ca an improved yolov5 based on the attention mechanism for mummy berry disease detection |
| topic | wild blueberry <i>Vaccinium angustifolium</i> <i>Monilinia vaccinii-corymbosi</i> deep learning coordinated attention synthetic data |
| url | https://www.mdpi.com/2077-0472/13/1/78 |
| work_keys_str_mv | AT efremyohannesobsie yolov5scaanimprovedyolov5basedontheattentionmechanismformummyberrydiseasedetection AT hongchunqu yolov5scaanimprovedyolov5basedontheattentionmechanismformummyberrydiseasedetection AT yongjiangzhang yolov5scaanimprovedyolov5basedontheattentionmechanismformummyberrydiseasedetection AT seannaannis yolov5scaanimprovedyolov5basedontheattentionmechanismformummyberrydiseasedetection AT francisdrummond yolov5scaanimprovedyolov5basedontheattentionmechanismformummyberrydiseasedetection |