摘要
针对荔枝果实个体小、生长密集和遮挡严重等特点,为了快速准确地实现荔枝的检测和计数,提出了一种融合注意力机制和多尺度特征图的网络模型。为了提高模型对遮挡和阴影环境下果实的识别准确率,将Coordinate Attention(CA)注意力机制嵌入至YOLOv4-Tiny模型。为了提高模型对小目标果实的检测精度,在特征金字塔Feature Pyramid Networks(FPN)结构中生成了两个更大尺度的特征图。试验结果表明:融合注意力机制的荔枝轻量化检测模型的准确率(Precision)、召回率(Recall)和平均精度(mAP)分别为92.92%、76.09%和88.51%。与YOLOv4-Tiny和YOLOv3模型相比,所构建的融合注意力机制的荔枝轻量化检测模型的平均检测精度分别高出8.84个百分点和3.87个百分点。该模型能够快速、精准地检测出果园环境中的荔枝,适用于果园中荔枝的识别和计数。
In view of the characteristic of small size,dense growth and severe occlusion of litchi,in order to rapidly and accurately detect and count litchi fruits,a network model combining attention mechanism and multi-scale feature maps was proposed in this study.For the purpose of improving the recognition accuracy of fruit in occluding and shaded environments,Coordinate Attention(CA)mechanism was embedded in YOLOv4-Tiny model.In order to improve the detection accuracy of the model for small target fruit,two larger scale feature maps were generated in the structure of Feature Pyramid Networks(FPN).The results showed that the Precision,Recall and mAP of the Litchi lightweight detection model combined with attention mechanism were 92.92%,76.09%and 88.51%,respectively.Compared with YOLOv4-Tiny and YOLOv3 models,the average detection accuracy of litchi lightweight detection model constructed in this paper with integrated attention mechanism was 8.84 percentage points and 3.87 percentage points higher,respectively.The model can detect litchi quickly and accurately in orchard environment,and suitable for identification and counting of litchi in orchard.
作者
王聪
文晟
兰玉彬
严倩
姜永华
张建桃
罗菊川
Wang Cong;Wen Sheng;Lan Yubin;Yan Qian;Jiang Yonghua;Zhang Jiantao;Luo Juchuan(College of Engineering,South China Agricultural University,Guangzhou 510642,China;National Joint Research Center of Precision Agriculture Aviation Application Technology,Guangzhou 510642,China;College of Electronic Engineering/College of Artificial Intelligence,South China Agricultural University,Guangzhou 510642,China;Fruit Research Institute,Guangdong Academy of Agricultural Sciences,Guangzhou 510645,China;College of Mathematics and Information,South China Agricultural University,Guangzhou 510642,China)
出处
《农机化研究》
北大核心
2025年第3期10-15,共6页
Journal of Agricultural Mechanization Research
基金
国家自然科学基金面上项目(32271985)
广东省自然科学基金面上项目(2022A1515011008)。
关键词
荔枝
注意力机制
特征金字塔
轻量化
检测方法
litchi
attention mechanism
feature pyramid network
lightweight
detection method
