中国农机化学报2024,Vol.45Issue(5) :195-201.DOI:10.13733/j.jcam.issn.2095-5553.2024.05.030

基于轻量化YOLOv4的死淘鸡目标检测算法

Dead chicken target detection algorithm based on lightweight YOLOv4

漆海霞 李承杰 黄桂珍
中国农机化学报2024,Vol.45Issue(5) :195-201.DOI:10.13733/j.jcam.issn.2095-5553.2024.05.030

基于轻量化YOLOv4的死淘鸡目标检测算法

Dead chicken target detection algorithm based on lightweight YOLOv4

漆海霞 1李承杰 2黄桂珍2
扫码查看

作者信息

  • 1. 华南农业大学工程学院,广州市,510642;国家精准农业航空施药技术国际联合研究中心,广州市,510642;岭南现代农业科学与技术广东省实验室,广州市,510642
  • 2. 华南农业大学工程学院,广州市,510642
  • 折叠

摘要

针对目前死淘鸡目标检测研究较少,高精度检测算法体积大难以部署至移动式设备等问题,提出一种基于YOLOv4的轻量化死淘鸡目标检测算法.采集大规模蛋鸡养殖工厂笼中死淘鸡图片,建立目标检测数据集;在算法中引入MobileNetv3主干提取网络与深度可分离卷积来降低模型体积;并在最大池化层前添加自注意力机制模块,增强算法对全局语义信息的捕获.在自建数据集中的试验结果表明,改进算法在死淘鸡目标检测任务中有更高的准确度,其mAP值与召回率分别达到97.74%和98.15%,模型大小缩小至原算法的1/5,在GPU加速下帧数达到77帧/s,检测速度提高1倍,能够满足嵌入式部署需求.

Abstract

Aiming at the problems that there are few studies on dead chicken target detection and the large size of the high-precision detection algorithm makes it difficult to deploy to mobile devices,a lightweight dead chicken target detection algorithm based on YOLOv4 is proposed.Firstly,the team collects images of dead chickens in cages from large-scale egg production plants to build a target detection dataset.Then,MobileNetv3 backbone extraction network with depth-separable convolution is introduced in the algorithm to reduce the model size.Finally,a self-attentive mechanism module is added before the maximum pooling layer to enhance the algorithm's capture of global semantic information.Experimental results in a self-built dataset show that the improved algorithm has higher accuracy in the dead pheasant target detection task,with mAP values and recall rates of 97.74%and 98.15%respectively.The model size is reduced to 1/5 of the original algorithm,and the frame rate reaches 77 frames/s under GPU acceleration,doubling the detection speed and meeting the requirements of embedded deployments.

关键词

死淘鸡识别/深度学习/轻量化网络/MobileNet/深度可分离卷积

Key words

identification of dead chicken/deep learning/lightweight network/MobileNet/deep separable convolution

引用本文复制引用

基金项目

广州市科技计划(20212100026)

出版年

2024
中国农机化学报
农业部南京农业机械化研究所

中国农机化学报

CSTPCD北大核心
影响因子:0.684
ISSN:2095-5553
参考文献量22
段落导航相关论文