Hierarchical Detection of <i>Gastrodia elata</i> Based on Improved YOLOX
Identifying the grade of <i>Gastrodia elata</i> in the market has low efficiency and accuracy. To address this issue, an I-YOLOX object detection algorithm based on deep learning and computer vision is proposed in this paper. First, six types of <i>Gastrodia elata</i> images...
| Main Authors: | , , , , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
MDPI AG
2023-05-01
|
| Series: | Agronomy |
| Subjects: | |
| Online Access: | https://www.mdpi.com/2073-4395/13/6/1477 |
| _version_ | 1797596532874149888 |
|---|---|
| author | Xingwei Duan Yuhao Lin Lixia Li Fujie Zhang Shanshan Li Yuxin Liao |
| author_facet | Xingwei Duan Yuhao Lin Lixia Li Fujie Zhang Shanshan Li Yuxin Liao |
| author_sort | Xingwei Duan |
| collection | DOAJ |
| description | Identifying the grade of <i>Gastrodia elata</i> in the market has low efficiency and accuracy. To address this issue, an I-YOLOX object detection algorithm based on deep learning and computer vision is proposed in this paper. First, six types of <i>Gastrodia elata</i> images of different grades in the <i>Gastrodia elata</i> planting cooperative were collected for image enhancement and labeling as the model training dataset. Second, to improve feature information extraction, an ECA attention mechanism module was inserted between the backbone network CSPDarknet and the neck enhancement feature extraction network FPN in the YOLOX model. Then, the impact of the attention mechanism and application position on model improvement was investigated. Third, the 3 × 3 convolution in the neck enhancement feature extraction network FPN and the head network was replaced by depthwise separable convolution (DS Conv) to reduce the model size and computation amount. Finally, the EIoU loss function was used to predict boundary frame regression at the output prediction end to improve the convergence speed of the model. The experimental results indicated that compared with the original YOLOX model, the mean average precision of the improved I-YOLOX network model was increased by 4.86% (97.83%), the model computation was reduced by 5.422 M (reaching 3.518 M), the model size was reduced by 20.6 MB (reaching 13.7 MB), and the image frames detected per second increased by 3 (reaching 69). Compared with other target detection algorithms, the improved model outperformed Faster R-CNN, SSD-VGG, YOLOv3s, YOLOv4s, YOLOv5s, and YOLOv7 algorithms in terms of mean average precision, model size, computation amount, and frames per second. The lightweight model improved the detection accuracy and speed of different grades of <i>Gastrodia elata</i> and provided a theoretical basis for the development of online identification systems of different grades of <i>Gastrodia elata</i> in practical production. |
| first_indexed | 2024-03-11T02:52:46Z |
| format | Article |
| id | doaj.art-5b05850391a142c8a95a05a214580ea7 |
| institution | Directory Open Access Journal |
| issn | 2073-4395 |
| language | English |
| last_indexed | 2024-03-11T02:52:46Z |
| publishDate | 2023-05-01 |
| publisher | MDPI AG |
| record_format | Article |
| series | Agronomy |
| spelling | doaj.art-5b05850391a142c8a95a05a214580ea72023-11-18T08:53:48ZengMDPI AGAgronomy2073-43952023-05-01136147710.3390/agronomy13061477Hierarchical Detection of <i>Gastrodia elata</i> Based on Improved YOLOXXingwei Duan0Yuhao Lin1Lixia Li2Fujie Zhang3Shanshan Li4Yuxin Liao5Faculty of Modern Agricultural Engineering, Kunming University of Science and Technology, Kunming 650500, ChinaFaculty of Modern Agricultural Engineering, Kunming University of Science and Technology, Kunming 650500, ChinaFaculty of Modern Agricultural Engineering, Kunming University of Science and Technology, Kunming 650500, ChinaFaculty of Modern Agricultural Engineering, Kunming University of Science and Technology, Kunming 650500, ChinaFaculty of Modern Agricultural Engineering, Kunming University of Science and Technology, Kunming 650500, ChinaFaculty of Electric Power Engineering, Kunming University of Science and Technology, Kunming 650500, ChinaIdentifying the grade of <i>Gastrodia elata</i> in the market has low efficiency and accuracy. To address this issue, an I-YOLOX object detection algorithm based on deep learning and computer vision is proposed in this paper. First, six types of <i>Gastrodia elata</i> images of different grades in the <i>Gastrodia elata</i> planting cooperative were collected for image enhancement and labeling as the model training dataset. Second, to improve feature information extraction, an ECA attention mechanism module was inserted between the backbone network CSPDarknet and the neck enhancement feature extraction network FPN in the YOLOX model. Then, the impact of the attention mechanism and application position on model improvement was investigated. Third, the 3 × 3 convolution in the neck enhancement feature extraction network FPN and the head network was replaced by depthwise separable convolution (DS Conv) to reduce the model size and computation amount. Finally, the EIoU loss function was used to predict boundary frame regression at the output prediction end to improve the convergence speed of the model. The experimental results indicated that compared with the original YOLOX model, the mean average precision of the improved I-YOLOX network model was increased by 4.86% (97.83%), the model computation was reduced by 5.422 M (reaching 3.518 M), the model size was reduced by 20.6 MB (reaching 13.7 MB), and the image frames detected per second increased by 3 (reaching 69). Compared with other target detection algorithms, the improved model outperformed Faster R-CNN, SSD-VGG, YOLOv3s, YOLOv4s, YOLOv5s, and YOLOv7 algorithms in terms of mean average precision, model size, computation amount, and frames per second. The lightweight model improved the detection accuracy and speed of different grades of <i>Gastrodia elata</i> and provided a theoretical basis for the development of online identification systems of different grades of <i>Gastrodia elata</i> in practical production.https://www.mdpi.com/2073-4395/13/6/1477<i>Gastrodia elata</i>YOLOXtarget detectionECADS ConvEIoU |
| spellingShingle | Xingwei Duan Yuhao Lin Lixia Li Fujie Zhang Shanshan Li Yuxin Liao Hierarchical Detection of <i>Gastrodia elata</i> Based on Improved YOLOX Agronomy <i>Gastrodia elata</i> YOLOX target detection ECA DS Conv EIoU |
| title | Hierarchical Detection of <i>Gastrodia elata</i> Based on Improved YOLOX |
| title_full | Hierarchical Detection of <i>Gastrodia elata</i> Based on Improved YOLOX |
| title_fullStr | Hierarchical Detection of <i>Gastrodia elata</i> Based on Improved YOLOX |
| title_full_unstemmed | Hierarchical Detection of <i>Gastrodia elata</i> Based on Improved YOLOX |
| title_short | Hierarchical Detection of <i>Gastrodia elata</i> Based on Improved YOLOX |
| title_sort | hierarchical detection of i gastrodia elata i based on improved yolox |
| topic | <i>Gastrodia elata</i> YOLOX target detection ECA DS Conv EIoU |
| url | https://www.mdpi.com/2073-4395/13/6/1477 |
| work_keys_str_mv | AT xingweiduan hierarchicaldetectionofigastrodiaelataibasedonimprovedyolox AT yuhaolin hierarchicaldetectionofigastrodiaelataibasedonimprovedyolox AT lixiali hierarchicaldetectionofigastrodiaelataibasedonimprovedyolox AT fujiezhang hierarchicaldetectionofigastrodiaelataibasedonimprovedyolox AT shanshanli hierarchicaldetectionofigastrodiaelataibasedonimprovedyolox AT yuxinliao hierarchicaldetectionofigastrodiaelataibasedonimprovedyolox |