Knowledge Distillation in YOLOX-ViT for Side-Scan Sonar Object Detection
arXiv (Cornell University)(2024)
摘要
In this paper we present YOLOX-ViT, a novel object detection model, andinvestigate the efficacy of knowledge distillation for model size reductionwithout sacrificing performance. Focused on underwater robotics, our researchaddresses key questions about the viability of smaller models and the impact ofthe visual transformer layer in YOLOX. Furthermore, we introduce a newside-scan sonar image dataset, and use it to evaluate our object detector'sperformance. Results show that knowledge distillation effectively reduces falsepositives in wall detection. Additionally, the introduced visual transformerlayer significantly improves object detection accuracy in the underwaterenvironment. The source code of the knowledge distillation in the YOLOX-ViT isat https://github.com/remaro-network/KD-YOLOX-ViT.
更多查看译文
关键词
Object Detection
AI 理解论文
溯源树
样例
生成溯源树,研究论文发展脉络
Chat Paper
正在生成论文摘要