基于改进YOLOv4的羊只检测方法研究

作     者：李远征 李章辉 王天一 LI Yuanzheng;LI Zhanghui;WANG Tianyi

作者机构：贵州大学大数据与信息工程学院贵阳550025 

出 版 物：《智能计算机与应用》 (Intelligent Computer and Applications)

年 卷 期：2023年第13卷第5期

页      面：175-180,186页

学科分类：1305[艺术学-设计学（可授艺术学、工学学位）] 13[艺术学] 081104[工学-模式识别与智能系统] 08[工学] 0804[工学-仪器科学与技术] 081101[工学-控制理论与控制工程] 0811[工学-控制科学与工程] 

基　　金：贵州省科学技术基金(黔科合基础-ZK一般304 黔科合基础1Y254) 

主　　题：羊只检测 YOLOv4 轻量化网络 注意力机制 非极大值抑制 

摘      要：羊只自动检测是大规模智能化羊养殖的基础。针对养殖场环境中存在围栏遮挡以及目标相互遮挡导致检测方法效率低、精度差和易漏检等问题,本文提出了一种基于改进YOLOv4的养殖场环境下羊只检测方法。采用轻量级网络ShuffleNet V2作为主干特征提取网络,使用深度可分离卷积替换普通卷积,在提升检测精度的同时将网络轻量化;引入注意力机制增强特征提取能力;在预测阶段使用DIoU-NMS提高检测精度。改进方法的检测精度达93.57%,检测速度达60 frame/s,参数量降低至41.13MB,能够有效提升养殖场环境下羊只检测的精度与速度。