河南农业科学2024，Vol.53Issue(5)：157-163,7.DOI:10.15933/j.cnki.1004-3268.2024.05.017
基于YOLOv5s-SE和通道剪枝的虫咬紫金蝉茶检测方法研究
Research on Insect-bitten Zijin Tea Detection Method Based on YOLOv5s-SE and Channel Pruning
摘要
Abstract
In order to achieve rapid and accurate identification of insect-bitten Zijin tea leaves in complex nature backgrounds,a detection method for Zijin tea based on YOLOv5s-SE and channel pruning was proposed.Firstly,SE modules were added to the backbone network of YOLOv5s to enhance the model's feature extraction capability and reduce interference from complex backgrounds during tea leaf feature extraction.Then,a channel pruning algorithm was used to prune the model and fine-tuning was conducted,enabling fast and accurate detection of insect-bitten Zijin tea leaves.Compared to YOLOv5s,the test results showed that the pruned model reduced parameters by 60.1％,improved FPS by 18.6％,reduced GFLOPs by 29.7％,and achieved mAP of 81.3％.关键词
茶叶/紫金蝉茶/目标检测/通道剪枝/注意力机制/深度学习Key words
Tea/Object detection/Channel pruning/Attention mechanism/Deep learning分类
农业科技引用本文复制引用
戴佳兵,宋春芳,凌彩金,李臻锋,孙崇高..基于YOLOv5s-SE和通道剪枝的虫咬紫金蝉茶检测方法研究[J].河南农业科学,2024,53(5):157-163,7.基金项目
以农产品为单元的广东省现代农业产业技术体系创新团队建设项目(茶叶)(2023KJ120) （茶叶）
河源市科技计划项目(河科2021030) （河科2021030）
2024紫金县科技计划项目 （）
