北京邮电大学学报

  • EI核心期刊

北京邮电大学学报

• •    

基于边缘智能的田间道路缺陷检测方法

陈泽宇1,古月2,程思瑶2,冯国惠3   

  1. 1. 哈尔滨工业大学计算机科学与技术学院
    2. 哈尔滨工业大学
    3. 北大荒信息有限公司
  • 收稿日期:2023-10-26 修回日期:2024-02-05 发布日期:2024-06-25
  • 通讯作者: 古月
  • 基金资助:
    科技创新2030-"新一代人工智能"重大项目;黑龙江省重点研发计划项目;国家自然科学基金面上项目;青年科学基金项目;青年科学基金项目

Edge Intelligence Based Field Road Defect Detection Method

  • Received:2023-10-26 Revised:2024-02-05 Published:2024-06-25
  • Contact: Yue GU

摘要: 在智慧农机的自主作业任务中,除自主导航任务外,还需要关注农机在自主作业时的田间道路缺陷检测问题。在田间坑洼检测任务中,使用Kaggle上公开的常规道路坑洼数据进行学习和训练,再对实验室环境下自建的农田坑洼数据集进行迁移学习,模拟实现田间道路中坑洼的检测。在模型设计方面,采用YOLOv5的轻量版本YOLOv5s进行网络设计,通过引入CBAM注意力机制来提升网络精度。最后,通过Jetson Nano作为载体,运用TensorRT对田间坑洼检测任务中所得到的模型进行边缘部署,在适当降低精度的情况下大幅提升了运行速度,满足了边缘端实时检测的要求,实现了农机自主作业任务中田间道路缺陷检测技术的边缘部署。

关键词: 农机自主作业, 缺陷检测, 深度学习, 轻量模型设计, 模型边缘部署

Abstract: Academic semantic translation using computers In the autonomous operation task of smart agricultural machinery, in addition to autonomous navigation tasks, it is also necessary to pay attention to the problem of field road defect detection during autonomous operation of agricultural machinery. In the task of detecting potholes in the field, we use the publicly available conventional road pothole data on Kaggle for learning and training, and then transfer learning from the self-built farmland pothole dataset in the laboratory environment to simulate the detection of potholes in the field road. In terms of model design, we adopt the lightweight version of YOLOv5s for network design, and introduce the CBAM attention mechanism to improve network accuracy. Finally, using Jetson Nano as a carrier, we apply TensorRT to simulate the detection of potholes in the field road. The model obtained from the pothole detection task is deployed on the edge, significantly improving the running speed while appropriately reducing the accuracy, meeting the requirements of real-time detection on the edge, and realizing the edge deployment of field road defect detection technology in autonomous agricultural machinery operation tasks.

Key words: Autonomous farm machinery operations, defect detection, deep learning, lightweight model design, model edge deployment

中图分类号: