14 December 2022 DRYDet: decoupling YOLOv5 detector and refining the tasks in remote sensing scenario
Wangming Zeng, Xiaohai He, Shuqi Xiong, Honggang Chen, Linbo Qing
Author Affiliations +
Abstract

You only look once version 5 (YOLOv5) algorithm uses one convolutional layer for both classification and localization tasks. They share the same set of weights, leading to interference between the two tasks. Thus, the application of this method in remote sensing scenario is very limited. To solve the problem, we decouple the detection head of the YOLOv5 algorithm so that these two tasks can be separated. Furthermore, we achieve high quality detection in remote sensing scene with task refinement. Specifically, we propose a weight allocation method based on Huffman coding theory to ease imbalance of the number of categories and improve the classification accuracy. If the number of a category is large, a small weight is set. Otherwise, a large weight is given. To improve the localization accuracy, we use the efficient intersection over union (IOU) loss to alleviate the problem of penalty failure caused by central IOU loss. In this way, the predicted boxes are closer to the ground truth. A large number of experiments were carried out on DOTA and UCAS-AOD datasets to verify the effectiveness of our proposed method. Compared with baseline, the mean average precision totally increased by 2.29% in DOTAv1.5 dataset. Simultaneously, our method also showed superior performance when comparing with start of the art algorithms.

© 2022 Society of Photo-Optical Instrumentation Engineers (SPIE)
Wangming Zeng, Xiaohai He, Shuqi Xiong, Honggang Chen, and Linbo Qing "DRYDet: decoupling YOLOv5 detector and refining the tasks in remote sensing scenario," Journal of Applied Remote Sensing 16(4), 046513 (14 December 2022). https://doi.org/10.1117/1.JRS.16.046513
Received: 24 April 2022; Accepted: 17 November 2022; Published: 14 December 2022
Lens.org Logo
CITATIONS
Cited by 2 scholarly publications.
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Remote sensing

Object detection

Head

Detection and tracking algorithms

Education and training

Visualization

Convolution

Back to Top