基于Jetson Nano+YOLO v5的哺乳期仔猪目标检测
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

基金项目:

江苏省科技计划项目(BE2019382)


Object Detection of Suckling Piglets Based on Jetson Nano and YOLO v5
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    针对仔猪个体小、易被遮挡且仔猪目标检测方法不易在嵌入式端部署等问题,提出一种适用于Jetson Nano端部署的哺乳期仔猪目标检测方法,在准确检测哺乳期仔猪目标的同时,使模型实地部署更加灵活。使用哺乳期仔猪图像建立数据集,数据量为14000幅,按8∶1∶1划分训练集、测试集和验证集。利用深度学习网络提取哺乳期仔猪特征,构建仔猪目标检测模型。融合推理网络中的Conv、BN、Activate Function层,合并相同维度张量,删除Concat层,实现网络结构量化,减少模型运行时的算力需求。将优化后模型迁移至Jetson Nano,在嵌入式平台进行测试。实验结果表明,在嵌入式端,量化后YOLO v5中4种模型的单帧图像平均运行时间分别为65、170、315、560ms,检测准确率分别为96.8%、97.0%、97.0%和96.6%,能够在Jetson Nano设备上对哺乳期仔猪目标实现精准检测,为仔猪目标检测的边缘计算模式奠定基础。

    Abstract:

    The deployment of piglet target detection model at the edge of the device is an important basis for fine management of piglets during lactation. Recognition of suckling piglets under complex environments is a difficult task, and deep learning methods are usually used to solve this problem. However, the object detection model of piglets based on deep learning often needs high computer force support, which is difficult to deploy in the field. To solve these problems, a object detection model of suckling piglets based on embedded terminal deployment was proposed, which made the deployment of piglet object detection system more flexible. A database was established by using images of suckling piglets with a data volume of 14000 pieces. The training set, test set, and validation set were divided by 8∶1∶1. The YOLO v5s, YOLO v5m, YOLO v5l, and YOLO v5x deep learning networks were trained to extract the characteristics of suckling piglets, and the corresponding piglets detection model was established to conduct object detection for suckling piglets. The Conv, BN, Activation Function layer, the same tensor and operation part of the network were fused, and the Concat layer was deleted to quantify the network structure and reduce the computational force demand of the model during operation. An embedded device Jetson Nano was used to infer the modified model to realize the deployment of piglet target detection model in the embedded terminal. The experimental results showed that the average running time of the optimized YOLO v5s, YOLO v5m, YOLO v5l, and YOLO v5x models were 65ms, 170ms, 315ms and 560ms, respectively, but the detection accuracy was dropped to 96.8%, 97.0%, 97.0% and 96.6%, respectively. The optimized YOLO v5s model can implement real-time detection of suckling piglets on embedded devices, which can lay a foundation for the edge computing model of piglets detection and provide technical support for precision breeding.

    参考文献
    相似文献
    引证文献
引用本文

丁奇安,刘龙申,陈佳,太猛,沈明霞.基于Jetson Nano+YOLO v5的哺乳期仔猪目标检测[J].农业机械学报,2022,53(3):277-284. DING Qi’an, LIU Longshen, CHEN Jia, TAI Meng, SHEN Mingxia. Object Detection of Suckling Piglets Based on Jetson Nano and YOLO v5[J]. Transactions of the Chinese Society for Agricultural Machinery,2022,53(3):277-284.

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2021-03-23
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2022-03-10
  • 出版日期: