To improve the accuracy of obstacle localization, a data-augmentation-based monocular visual obstacle localization method for UAV is proposed in this paper. The method can be divided into two parts: obstacle detection and obstacle localization. In the first part, the obstacle detection algorithm based on the YOLOv5 network is trained. To solve the problem of difficulty in obtaining samples, the data augmentation method is used to simulate different flight environments and camera equipment of UAV. To enhance the accuracy of detection, add penalty terms for the distance between the detection box and the ground truth box center, as well as the aspect ratio of the two boxes, to the loss function of the detection algorithm. In obstacle localization, according to the conversion formula from pixel coordinates to ground coordinates, the location of the obstacle can be achieved with calibrated intrinsic parameters of the camera and the UAV’s attitude. Consequently, UAV can recognize and locate obstacles and take action to avoid collision.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Data-Augmentation-Based Monocular Visual Obstacle Localization Method for UAV

  • Yanjun Wang,
  • Teng Long,
  • Jianxin Zhong,
  • Zeyang Xie,
  • Jingliang Sun,
  • Zhenlin Zhou

摘要

To improve the accuracy of obstacle localization, a data-augmentation-based monocular visual obstacle localization method for UAV is proposed in this paper. The method can be divided into two parts: obstacle detection and obstacle localization. In the first part, the obstacle detection algorithm based on the YOLOv5 network is trained. To solve the problem of difficulty in obtaining samples, the data augmentation method is used to simulate different flight environments and camera equipment of UAV. To enhance the accuracy of detection, add penalty terms for the distance between the detection box and the ground truth box center, as well as the aspect ratio of the two boxes, to the loss function of the detection algorithm. In obstacle localization, according to the conversion formula from pixel coordinates to ground coordinates, the location of the obstacle can be achieved with calibrated intrinsic parameters of the camera and the UAV’s attitude. Consequently, UAV can recognize and locate obstacles and take action to avoid collision.