中国农业机械化科学研究院集团有限公司 主管

北京卓众出版有限公司 主办

翁嘉杰,毛战华.基于深度学习的家畜虚拟电子围栏设计[J].农业工程,2024,14(4):41-45. DOI: 10.19998/j.cnki.2095-1795.2024.04.007
引用本文: 翁嘉杰,毛战华.基于深度学习的家畜虚拟电子围栏设计[J].农业工程,2024,14(4):41-45. DOI: 10.19998/j.cnki.2095-1795.2024.04.007
WENG Jiajie,MAO Zhanhua.Design of livestock virtual electronic fence based on deep learning[J].Agricultural Engineering,2024,14(4):41-45. DOI: 10.19998/j.cnki.2095-1795.2024.04.007
Citation: WENG Jiajie,MAO Zhanhua.Design of livestock virtual electronic fence based on deep learning[J].Agricultural Engineering,2024,14(4):41-45. DOI: 10.19998/j.cnki.2095-1795.2024.04.007

基于深度学习的家畜虚拟电子围栏设计

Design of Livestock Virtual Electronic Fence Based on Deep Learning

  • 摘要: 针对牧场中家畜越界及传统电子围栏难以防止家畜越界的问题,基于改进后的YOLOv5算法,设计了虚拟电子围栏。采用YOLOv5s算法作为基础,并进行了迁移学习和添加ECA注意力模块。利用PyTorch框架进行训练,并对模型进行了评估,相比原版YOLOv5s,改进YOLOv5s算法对黄牛的检测精确率、召回率、平均精度均值分别提升0.2、1.3、0.7个百分点,单帧推理总耗时下降0.5 ms。将改进后的模型转换为RKNN格式,并部署在带有NPU的RK3588开发板上,加快模型推理速度。结果表明,应用深度学习技术与ROI划定技术,成功设计了家畜虚拟电子围栏,优化智慧牧场的管理体系,提高管理效率,降低管理成本,具备一定的实用价值。

     

    Abstract: In response to issue of livestock crossing boundaries in pastures and limitations of traditional electronic fences in preventing such occurrences, a virtual electronic fence was designed based on improved version of YOLOv5.Initially, YOLOv5s model was used as foundation, and transfer learning and addition of an ECA attention module were performed.Subsequently, the model was trained using PyTorch framework and evaluated.Compared to original YOLOv5s, improved YOLOv5s achieved 0.2, 1.3, and 0.7 percentage points increase in precision, recall, and mAP for cattle detection, respectively, while reducing single-frame inference total time by 0.5 ms.Finally, improved model was converted to RKNN format and deployed on an NPU-equipped RK3588 development board to accelerate model's inference speed.Results showed that, application of deep learning technology and ROI delineation technology has successfully designed a virtual electronic fence for livestock, optimized management system of smart ranches, improved management efficiency, reduced management costs, and has certain practical value.

     

/

返回文章
返回