关键词:
计算机视觉
生猪姿态识别
YOLOv8s
多目标跟踪
BoT-SORT
摘要:
随着我国乡村振兴战略的全面推进,养猪业迅速发展,养猪业成为我国农业领域的重要组成部分,在成本管控、技术运用、市场应对及政策导向的作用下,养猪行业的规模化养殖得到了快速发展,规模化养殖慢慢占据主导地位。在规模化养殖中,疾病是制约我国生猪饲养产业持续发展的关键因素。然而,如何提前发现和预防疾病,就显得十分重要。相关研究显示,生猪在生长过程的姿态变化,往往是疾病出现的先兆,群猪姿态识别与跟踪获取的生猪数据能成为生猪疾病诊断和预防的重要参考。凭借这些数据,可提前对生猪健康状况做出诊断,有效预防常见疾病,防止因大规模感染带来经济损失,有力推动我国养猪行业的进步。
本文以群养生猪图像和视频作为研究对象,将结构重参数化、可变核卷积以及多尺度卷积注意力机制与卷积神经神经网络进行结合,构建生猪姿态识别模型,以改进后的姿态识别模型RMAK-YOLOv8s为检测模型,与BoT-SORT跟踪算法相结合,替换轻量化的重识别网络Mobile Net V3,在保证准确率前提下提高检测和跟踪速度,实现有效的生猪姿态识别和跟踪,为后续生猪疾病的预防和行为分析提供有效的数据支持。论文的主要研究内容如下:
(1)生猪姿态数据集和生猪跟踪数据集的构建。鉴于生猪公开数据集匮乏的状况,通过安装在养殖场的高清摄像头并配合人工拍照,成功采集到了生猪呈现出站立、坐立、躺卧以及跪立这四种不同姿态的图像,以及包含生猪相关活动的视频资料。在预处理阶段,先是从部分视频里提取关键帧,接着对图像数据开展人工筛选工作。筛选完成后,进行数据集的划分操作,并且运用图像增强技术,实现对生猪姿态图像数据集的扩充。最后使用Labelme标注工具完成标注工作,对处理后的生猪姿态数据加以标注,得到最终的生猪姿态数据集。选取7个实验视频作为实验对象,使用Dark Label软件进行数据标注,自定义数据标注格式,构建跟踪数据集,为后续多目标跟踪的性能评价奠定基础。
(2)研究基于改进YOLOv8s的生猪姿态识别方法。针对复杂场景下生猪姿态识别效果差,现有生猪检测算法参数量多和计算量大,不利于实地部署和后续及时跟踪等问题,本文以YOLOv8s为基准网络,改进C2f模块,实现隐式特征复用;添加多尺度卷积注意力机制,用于捕捉多尺度特征图,加强有效特征的权重比例;使用可变核卷积代替标准卷积,获得更有效的特征信息,为网络开销和性能之间的权衡提供更多的选择。在此基础上构建了基于改进YOLOv8s的生猪姿态识别模型即RMAK-YOLOv8s模型。实验表明,改进后的RMAK-YOLOv8s模型参数使用量减少10.77%,计算量减少5.22%,改进模型的m AP(0.5)、m AP(0.5:0.95)达到了93.7%、78.5%,分别提升1.7个百分点和1.3个百分点。结果表明,本文所提出的模型能更好地识别生猪姿态,为后续跟踪、行为分析提供模型支持。
(3)研究基于改进BoT-SORT的生猪多目标跟踪方法。针对计算资源的高需求和实时性的挑战,利用改进后的RMAK-YOLOv8s模型并结合BoT-SORT算法作为生猪姿态识别和跟踪的基本框架,同时替换Res Ne St50算法选择更为轻量化的Mobile Net V3算法作为重识别网络,提出一种轻量化的跟踪算法,降低计算成本,实验结果表明,在多目标跟踪方面,BoT-SORT算法与Byte SORT算法和Strong SORT算法相比,MOTA和IDF1表现最佳,跟踪效果好,此外,所提方法使MOTA和IDF1分别提升5个百分点和2.9个百分点,平均跟踪速度从25.6帧/s提升到42.3帧/s,经算法改良,从理论层面达成了实时跟踪的可行性,使得该算法模型更契合部署于边缘设备之上,显著提升了复杂场景下多目标跟踪性能,为长期行为分析和非接触式自动监控提供有力支持。