农业机械学报

作者：漆海霞，冯发生，尹选春，杨泽康，周子森，梁广升

单位：华南农业大学,华南农业大学,华南农业大学,华南农业大学,华南农业大学,华南农业大学,

关键词：猪只行为识别;模型轻量化;通道剪枝;知识蒸馏;YOLO v5n;嵌入式设备

基金：广州市农村科技特派员项目（20212100026）和特定高校学科建设项目（2023B10564002）

针对现有猪只行为识别模型体积大、识别场景单一、部署应用硬件要求高等问题，本文提出轻量级多场景群养猪只行为识别模型YOLO v5n-PBR（YOLO v5n for pig behavior recognition）。首先通过拍摄和收集不同养殖场景、不同猪只数量及不同角度的群养猪只行为数据构建多场景群养猪只行为数据集，并根据该数据集中猪只行为目标的特点引入迁移学习方法和OTA（Optimal transport assignment）标签分配方法对YOLO v5n模型进行训练，加快模型收敛速度并提升模型精度，构建高精度多场景群养猪只行为识别模型;然后利用L1-norm剪枝算法筛选并删减模型中不重要的通道，去除冗余参数;最后通过微调训练和中间特征知识蒸馏去除剪枝带来的性能劣化，从而得到轻量级多场景群养猪只行为识别模型YOLO v5n-PBR并进行嵌入式设备部署。实验结果表明，YOLO v5n-PBR模型平均精度均值 (mean average precision，mAP)为96.9%，参数量、计算量和内存占用量分别为4.700×105、1.20×109和1.2 MB，在两种不同系统和不同硬件配置的嵌入式设备上的部署实时识别帧率分别为12.2帧/s和66.3帧/s，与原始模型YOLO v5n相比，mAP提高1.1个百分点，参数量、计算量和内存占用量分别减少73.3%、70.7%和68.4%，部署实时识别帧率分别提高74.3%和83.1%。此外，基于多场景群养猪只行为数据集训练得到的YOLO v5n-PBR模型在4个单场景或双场景的群养猪只行为数据集上的mAP均能达到98.1%，对2种不同养殖场景的6段猪只行为视频的嵌入式设备部署识别统计结果与人工统计结果相近，平均精确率和平均召回率均为95.3%，以较少的参数达到较强的泛化性。本文提出的YOLO v5n-PBR模型具有精度高、体积小、速度快、泛化性强等优点，可满足嵌入式设备部署要求，为猪只行为的实时、准确监测及猪只行为识别模型的部署应用提供技术基础。

来源：2024年第10期

《农业机械学报》期刊编辑部

查看农业机械学报杂志2024年第10期