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

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

扫码查看
针对牧场中家畜越界及传统电子围栏难以防止家畜越界的问题,基于改进后的YOLOv5算法,设计了虚拟电子围栏.采用YOLOv5s算法作为基础,并进行了迁移学习和添加ECA注意力模块.利用PyTorch框架进行训练,并对模型进行了评估,相比原版YOLOv5s,改进YOLOv5s算法对黄牛的检测精确率、召回率、平均精度均值分别提升 0.2、1.3、0.7个百分点,单帧推理总耗时下降0.5 ms.将改进后的模型转换为RKNN格式,并部署在带有NPU的RK3588开发板上,加快模型推理速度.结果表明,应用深度学习技术与ROI划定技术,成功设计了家畜虚拟电子围栏,优化智慧牧场的管理体系,提高管理效率,降低管理成本,具备一定的实用价值.
Design of Livestock Virtual Electronic Fence Based on Deep Learning
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 us-ing 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 acceler-ate model's inference speed.Results showed that,application of deep learning technology and ROI delineation technology has success-fully designed a virtual electronic fence for livestock,optimized management system of smart ranches,improved management efficiency,reduced management costs,and has certain practical value.

YOLOv5RK3588deep learningvirtual fenceregion of interestsmart rancheslivestock

翁嘉杰、毛战华

展开 >

青岛农业大学理学与信息科学学院,山东 青岛 266000

YOLOv5 RK3588 深度学习 虚拟围栏 感兴趣区域 智慧牧场 家畜

2024

农业工程
北京卓众出版有限公司

农业工程

CSTPCD
影响因子:0.422
ISSN:2095-1795
年,卷(期):2024.14(4)
  • 18