河南农业科学2024,Vol.53Issue(4):161-171,11.DOI:10.15933/j.cnki.1004-3268.2024.04.018
基于SR-YOLOv8n-BCG的模糊花卉图像检测
Blurry Flower Image Detection Based on SR-YOLOv8n-BCG
摘要
Abstract
In order to meet the fast and accurate detection requirements for fuzzy flower images in complex environments,a composite model SR-YOLOv8n-BCG was proposed.This model effectively integrated the image reconstruction capability of SRGAN(super-resolution generative adversarial network)and the object detection capability of YOLOv8.Furthermore,the network structure was further improved to enhance accuracy and achieve lightweight design.Firstly,SR-YOLOv8n-BCG employed SRGAN to perform super-resolution processing on fuzzy flower images,enhancing the image quality input into model.Secondly,in the feature extraction network of YOLOv8n,a weighted bidirectional feature pyramid network(BiFPN)was used to replace the PAN-FPN module,effectively fusing multi-scale flower features and reducing the model size.Additionally,a coordinate attention(CA)mechanism was introduced to enhance the model's feature extraction ability.Finally,Ghost convolution was used to replace regular convolution,further improving detection accuracy and model lightweightness.Experimental results on a self-made dataset of five flower categories demonstrated that compared to SR-YOLOv8n,the SR-YOLOv8n-BCG model achieved an average precision improvement of 1.2 percentage points,reaching 95.4%with a 35.5%reduction in model size.The results above indicate that the proposed improved model effectively enhances the detection accuracy of fuzzy flower images and achieves lightweight design to suit lower-end devices.关键词
花卉/图像检测/SRGAN/YOLOv8/BiFPN/坐标注意力机制/Ghost卷积Key words
Flower/Image detection/SRGAN/YOLOv8/BiFPN/Coordinate attention mechanism/Ghost convolution分类
农业科技引用本文复制引用
黄小龙,陈中举,许浩然,李和平..基于SR-YOLOv8n-BCG的模糊花卉图像检测[J].河南农业科学,2024,53(4):161-171,11.基金项目
湖北省教育厅科学技术研究项目(B2021052) (B2021052)