Unmanned aerial vehicle (UAV) object detection algorithm based on keypoints representation and rotated distance-IoU loss

被引:3
|
作者
Zhu, Hufei [1 ]
Huang, Yonghui [1 ]
Xu, Ying [1 ]
Zhou, Jianhong [1 ]
Deng, Fuqin [1 ]
Zhai, Yikui [1 ]
机构
[1] Wuyi Univ, Sch Elect & Informat Engn, Jiangmen, Guangdong, Peoples R China
关键词
Key points represent rotated bounding boxes; Rotated distance-IoU loss; Object detection; Unmanned aerial vehicle; Deep learning;
D O I
10.1007/s11554-024-01444-6
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Recently, significant progress has been made in the research field of unmanned aerial vehicle (UAV) object detection through deep learning. The proliferation of unmanned aerial vehicles has notably facilitated the acquisition of corresponding data. However, the presence of substantial rotated objects in various orientations within UAV data sets poses challenges for traditional horizontal box object detection methods. These conventional approaches struggle to precisely locate rotated objects. Consequently, algorithms for rotated bounding-box object detection have been proposed; however, some of these existing methods exhibit issues, including periodicity of angle and exchangeability of edges. We propose a joint key point representation and rotated distance loss object detection network to solve the above problems. It is mainly composed of the key point representation module and the rotated distance-IoU loss. The key-points representation is used to indirectly represent the angle parameter of the rotated bounding box. It accomplishes this by measuring the angle between the line connecting the center point of the rotated bounding box to a specific boundary center point and the horizontal line. Next, the coordinates of the center points of anchor and the center points of its boundary are used to obtain the height dimension of the rotated bounding box and the width dimension of a rotated bounding box is introduced. Like this, the rotated bounding box can be represented by two points and a width dimension. Also, based on the traditional rotated IoU loss which does not incorporate the distance between the center point of the prediction box and the center point of ground truth in the regression process, the rotated distance-IoU loss is proposed to replace the traditional rotated IoU loss, which speeds up the convergence of the network. We have conducted extensive experiments on the DOTA data set and the DroneVehicle data set and have demonstrated the effectiveness of the proposed method.
引用
收藏
页数:14
相关论文
共 50 条
  • [1] Unmanned aerial vehicle (UAV) object detection algorithm based on keypoints representation and rotated distance-IoU loss
    Hufei Zhu
    Yonghui Huang
    Ying Xu
    Jianhong Zhou
    Fuqin Deng
    Yikui Zhai
    Journal of Real-Time Image Processing, 2024, 21
  • [2] Object Detection for UAV Aerial Scenarios Based on Vectorized IOU
    Lu, Shun
    Lu, Hanyu
    Dong, Jun
    Wu, Shuang
    SENSORS, 2023, 23 (06)
  • [3] Target Object Detection from Unmanned Aerial Vehicle (UAV) Images Based on Improved YOLO Algorithm
    Jawaharlalnehru, Arunnehru
    Sambandham, Thalapathiraj
    Sekar, Vaijayanthi
    Ravikumar, Dhanasekar
    Loganathan, Vijayaraja
    Kannadasan, Raju
    Khan, Arfat Ahmad
    Wechtaisong, Chitapong
    Haq, Mohd Anul
    Alhussen, Ahmed
    Alzamil, Zamil S.
    ELECTRONICS, 2022, 11 (15)
  • [4] UAV-YOLO: Small Object Detection on Unmanned Aerial Vehicle Perspective
    Liu, Mingjie
    Wang, Xianhao
    Zhou, Anjian
    Fu, Xiuyuan
    Ma, Yiwei
    Piao, Changhao
    SENSORS, 2020, 20 (08)
  • [5] A Real-Time Unmanned Aerial Vehicle (UAV) Aerial Image Object Detection Model
    Tan, Li
    Liu, Zikang
    Liu, He
    Li, Dongfang
    Zhang, Chen
    2024 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS, IJCNN 2024, 2024,
  • [6] YOLOv7-UAV: An Unmanned Aerial Vehicle Image Object Detection Algorithm Based on Improved YOLOv7
    Zeng, Yalin
    Zhang, Tian
    He, Weikai
    Zhang, Ziheng
    ELECTRONICS, 2023, 12 (14)
  • [7] UAV Vehicle Object Detection Algorithm Based on Efficientnet
    Jiang, Degang
    Jiang, Zhi
    Huang, Zijie
    Guo, Cailing
    Li, Bailin
    Computer Engineering and Applications, 2023, 59 (12) : 228 - 234
  • [8] OBJECT BASED CLASSIFICATION OF UNMANNED AERIAL VEHICLE (UAV) IMAGERY FOR FOREST FIRES MONITORING
    Bilgilioglu, B. Baha
    Ozturk, Ozan
    Sariturk, Batuhan
    Seker, Dursun Zafer
    FRESENIUS ENVIRONMENTAL BULLETIN, 2019, 28 (02): : 1011 - 1017
  • [9] Application of Deep Learning Based Object Detection on Unmanned Aerial Vehicle
    Ipek, Burak
    Akpinar, Mustafa
    2020 5TH INTERNATIONAL CONFERENCE ON COMPUTER SCIENCE AND ENGINEERING (UBMK), 2020, : 74 - 78
  • [10] Research on detection method of pavement diseases based on Unmanned Aerial Vehicle (UAV)
    Mao, Zhijian
    Zhao, Chihang
    Zheng, Youfeng
    Mao, Yan
    Li, Hao
    Hua, Liru
    Liu, Yang
    2020 INTERNATIONAL CONFERENCE ON IMAGE, VIDEO PROCESSING AND ARTIFICIAL INTELLIGENCE, 2020, 11584