RDR-KD: A Knowledge Distillation Detection Framework for Drone Scenes

Jinxiang Huang, Hong Chang, Xin Yang*, Yong Liu, Shuqi Liu, Yong Song*

*此作品的通讯作者

科研成果: 期刊稿件文章同行评审

摘要

Drone object detection (DOD) with real-time deployment is a research hotspot. On the one hand, the performance of tiny object detection is closely related to the ground detection capability of the drone platform. Existing methods are keen on designing complex networks to enhance the accuracy of tiny objects, which significantly increases computational costs. On the other hand, the limited drone hardware resources urgently require lightweight models for deployment. To address the dilemma of balancing detection accuracy and computational efficiency, we propose a regenerated-decoupled-responsive knowledge distillation (RDR-KD) framework specifically for drone scenes. First, we design the Regenerated Distillation and the Decoupled Distillation to fully transfer the tiny object feature information from the teacher model to the student model. Meanwhile, we devise the logit-based Responsive Distillation based on focal loss and efficient intersection over union (EIoU) to alleviate class imbalance. Finally, we conduct extensive experiments on the VisDrone2019 dataset. The experimental results demonstrate that the proposed RDR-KD framework improves AP and {mathrm{AP}}-{S} of the student model by 3.3% and 2.9% respectively, which outperforms other state-of-the-art distillation frameworks.

源语言英语
文章编号6008705
页(从-至)1-5
页数5
期刊IEEE Geoscience and Remote Sensing Letters
21
DOI
出版状态已出版 - 2024

指纹

探究 'RDR-KD: A Knowledge Distillation Detection Framework for Drone Scenes' 的科研主题。它们共同构成独一无二的指纹。

引用此