A Maturity Detection Method for Hemerocallis Citrina Baroni Based on Lightweight and Attention Mechanism

Hemerocallis citrina Baroni with different maturity levels has different uses for food and medicine and has different economic benefits and sales value. However, the growth speed of Hemerocallis citrina Baroni is fast, the harvesting cycle is short, and the maturity identification is completely depe...

Полное описание

Библиографические подробности
Главные авторы: Bin Sheng, Ligang Wu, Nan Zhang
Формат: Статья
Язык:English
Опубликовано: MDPI AG 2023-11-01
Серии:Applied Sciences
Предметы:
Online-ссылка:https://www.mdpi.com/2076-3417/13/21/12043
_version_ 1827765778285330432
author Bin Sheng
Ligang Wu
Nan Zhang
author_facet Bin Sheng
Ligang Wu
Nan Zhang
author_sort Bin Sheng
collection DOAJ
description Hemerocallis citrina Baroni with different maturity levels has different uses for food and medicine and has different economic benefits and sales value. However, the growth speed of Hemerocallis citrina Baroni is fast, the harvesting cycle is short, and the maturity identification is completely dependent on experience, so the harvesting efficiency is low, the dependence on manual labor is large, and the identification standard is not uniform. In this paper, we propose a GCB YOLOv7 Hemerocallis citrina Baroni maturity detection method based on a lightweight neural network and attention mechanism. First, lightweight Ghost convolution is introduced to reduce the difficulty of feature extraction and decrease the number of computations and parameters of the model. Second, between the feature extraction backbone network and the feature fusion network, the CBAM mechanism is added to perform the feature extraction independently in the channel and spatial dimensions, which improves the tendency of the feature extraction and enhances the expressive ability of the model. Last, in the feature fusion network, Bi FPN is used instead of the concatenate feature fusion method, which increases the information fusion channels while decreasing the number of edge nodes and realizing cross-channel information fusion. The experimental results show that the improved GCB YOLOv7 algorithm reduces the number of parameters and floating-point operations by about 2.03 million and 7.3 G, respectively. The training time is reduced by about 0.122 h, and the model volume is compressed from 74.8 M to 70.8 M. In addition, the average precision is improved from 91.3% to 92.2%, mAP@0.5 and mAP@0.5:0.95 are improved by about 1.38% and 0.20%, respectively, and the detection efficiency reaches 10 ms/frame, which meets the real-time performance requirements. It can be seen that the improved GCB YOLOv7 algorithm is not only lightweight but also effectively improves detection precision.
first_indexed 2024-03-11T11:32:41Z
format Article
id doaj.art-790931cd1c9248c294c42f22117801a5
institution Directory Open Access Journal
issn 2076-3417
language English
last_indexed 2024-03-11T11:32:41Z
publishDate 2023-11-01
publisher MDPI AG
record_format Article
series Applied Sciences
spelling doaj.art-790931cd1c9248c294c42f22117801a52023-11-10T14:59:40ZengMDPI AGApplied Sciences2076-34172023-11-0113211204310.3390/app132112043A Maturity Detection Method for Hemerocallis Citrina Baroni Based on Lightweight and Attention MechanismBin Sheng0Ligang Wu1Nan Zhang2College of Mechanical and Electrical Engineering, Shanxi Datong University, Datong 037003, ChinaCollege of Mechanical and Electrical Engineering, Shanxi Datong University, Datong 037003, ChinaStudent Work Department, Shanxi General Aviation Polytechnic, Datong 037003, ChinaHemerocallis citrina Baroni with different maturity levels has different uses for food and medicine and has different economic benefits and sales value. However, the growth speed of Hemerocallis citrina Baroni is fast, the harvesting cycle is short, and the maturity identification is completely dependent on experience, so the harvesting efficiency is low, the dependence on manual labor is large, and the identification standard is not uniform. In this paper, we propose a GCB YOLOv7 Hemerocallis citrina Baroni maturity detection method based on a lightweight neural network and attention mechanism. First, lightweight Ghost convolution is introduced to reduce the difficulty of feature extraction and decrease the number of computations and parameters of the model. Second, between the feature extraction backbone network and the feature fusion network, the CBAM mechanism is added to perform the feature extraction independently in the channel and spatial dimensions, which improves the tendency of the feature extraction and enhances the expressive ability of the model. Last, in the feature fusion network, Bi FPN is used instead of the concatenate feature fusion method, which increases the information fusion channels while decreasing the number of edge nodes and realizing cross-channel information fusion. The experimental results show that the improved GCB YOLOv7 algorithm reduces the number of parameters and floating-point operations by about 2.03 million and 7.3 G, respectively. The training time is reduced by about 0.122 h, and the model volume is compressed from 74.8 M to 70.8 M. In addition, the average precision is improved from 91.3% to 92.2%, mAP@0.5 and mAP@0.5:0.95 are improved by about 1.38% and 0.20%, respectively, and the detection efficiency reaches 10 ms/frame, which meets the real-time performance requirements. It can be seen that the improved GCB YOLOv7 algorithm is not only lightweight but also effectively improves detection precision.https://www.mdpi.com/2076-3417/13/21/12043deep learningneural networkchannel information fusionintelligent agriculture
spellingShingle Bin Sheng
Ligang Wu
Nan Zhang
A Maturity Detection Method for Hemerocallis Citrina Baroni Based on Lightweight and Attention Mechanism
Applied Sciences
deep learning
neural network
channel information fusion
intelligent agriculture
title A Maturity Detection Method for Hemerocallis Citrina Baroni Based on Lightweight and Attention Mechanism
title_full A Maturity Detection Method for Hemerocallis Citrina Baroni Based on Lightweight and Attention Mechanism
title_fullStr A Maturity Detection Method for Hemerocallis Citrina Baroni Based on Lightweight and Attention Mechanism
title_full_unstemmed A Maturity Detection Method for Hemerocallis Citrina Baroni Based on Lightweight and Attention Mechanism
title_short A Maturity Detection Method for Hemerocallis Citrina Baroni Based on Lightweight and Attention Mechanism
title_sort maturity detection method for hemerocallis citrina baroni based on lightweight and attention mechanism
topic deep learning
neural network
channel information fusion
intelligent agriculture
url https://www.mdpi.com/2076-3417/13/21/12043
work_keys_str_mv AT binsheng amaturitydetectionmethodforhemerocalliscitrinabaronibasedonlightweightandattentionmechanism
AT ligangwu amaturitydetectionmethodforhemerocalliscitrinabaronibasedonlightweightandattentionmechanism
AT nanzhang amaturitydetectionmethodforhemerocalliscitrinabaronibasedonlightweightandattentionmechanism
AT binsheng maturitydetectionmethodforhemerocalliscitrinabaronibasedonlightweightandattentionmechanism
AT ligangwu maturitydetectionmethodforhemerocalliscitrinabaronibasedonlightweightandattentionmechanism
AT nanzhang maturitydetectionmethodforhemerocalliscitrinabaronibasedonlightweightandattentionmechanism