arXiv reaDer
SSDA-YOLO: クロスドメイン オブジェクト検出のための半教師付きドメイン適応 YOLO
SSDA-YOLO: Semi-supervised Domain Adaptive YOLO for Cross-Domain Object Detection
ドメイン適応オブジェクト検出 (DAOD) は、クロスドメインの不一致によって引き起こされる転送パフォーマンスの低下を軽減することを目的としています。ただし、ほとんどの既存の DAOD メソッドは、時代遅れで計算量の多い 2 段階の Faster R-CNN によって支配されており、産業用アプリケーションの最初の選択肢ではありません。この論文では、コンパクトな1段階の強力な検出器YOLOv5をドメイン適応と統合することにより、クロスドメイン検出パフォーマンスを改善するための新しい半教師付きドメイン適応YOLO(SSDA-YOLO)ベースの方法を提案します。具体的には、ナレッジ蒸留フレームワークを Mean Teacher モデルに適応させて、学生モデルがラベル付けされていないターゲット ドメインのインスタンス レベルの機能を取得できるようにします。また、シーン スタイルの転送を利用して、異なるドメインで疑似画像をクロス生成し、画像レベルの違いを修正します。さらに、クロスドメイン予測をさらに調整するために、直感的な一貫性の喪失が提案されています。 PascalVOC、Clipart1k、Cityscapes、Foggy Cityscapes などの公開ベンチマークで SSDA-YOLO を評価します。さらに、その一般化を検証するために、さまざまな実際の教室から収集されたあくび検出データセットで実験を行います。結果は、これらの DAOD タスクでの方法の大幅な改善を示しており、提案された適応モジュールの有効性と、DAOD でより高度な検出器を適用する緊急性の両方を明らかにしています。コードは https://github.com/hnuzhy/SSDA-YOLO で入手できます。
Domain adaptive object detection (DAOD) aims to alleviate transfer performance degradation caused by the cross-domain discrepancy. However, most existing DAOD methods are dominated by outdated and computationally intensive two-stage Faster R-CNN, which is not the first choice for industrial applications. In this paper, we propose a novel semi-supervised domain adaptive YOLO (SSDA-YOLO) based method to improve cross-domain detection performance by integrating the compact one-stage stronger detector YOLOv5 with domain adaptation. Specifically, we adapt the knowledge distillation framework with the Mean Teacher model to assist the student model in obtaining instance-level features of the unlabeled target domain. We also utilize the scene style transfer to cross-generate pseudo images in different domains for remedying image-level differences. In addition, an intuitive consistency loss is proposed to further align cross-domain predictions. We evaluate SSDA-YOLO on public benchmarks including PascalVOC, Clipart1k, Cityscapes, and Foggy Cityscapes. Moreover, to verify its generalization, we conduct experiments on yawning detection datasets collected from various real classrooms. The results show considerable improvements of our method in these DAOD tasks, which reveals both the effectiveness of proposed adaptive modules and the urgency of applying more advanced detectors in DAOD. Our code is available on https://github.com/hnuzhy/SSDA-YOLO.
updated: Sun Nov 27 2022 09:23:35 GMT+0000 (UTC)
published: Fri Nov 04 2022 01:50:13 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト