arXiv reaDer
ハイブリッド マッチングによる DETR
DETRs with Hybrid Matching
1 対 1 のセット マッチングは、DETR がエンド ツー エンド機能を確立するための重要な設計であるため、オブジェクト検出では、重複した検出を削除するために手作りの NMS (非最大抑制) を必要としません。このエンド ツー エンドの署名は、DETR の汎用性にとって重要であり、より広い視野のタスクに一般化されています。ただし、ポジティブ サンプルとして割り当てられたクエリはほとんどなく、1 対 1 のセット マッチングによってポジティブ サンプルのトレーニング効果が大幅に低下することに注意してください。トレーニング中に元の 1 対 1 のマッチング ブランチと補助的な 1 対多のマッチング ブランチを組み合わせるハイブリッド マッチング スキームに基づく、シンプルかつ効果的な方法を提案します。私たちのハイブリッド戦略は、精度を大幅に向上させることが示されています。推論では、元の 1 対 1 の一致分岐のみが使用されるため、エンドツーエンドのメリットと DETR と同じ推論効率が維持されます。この方法は H-DETR と名付けられており、DeformableDETR、PETRv2、PETR、TransTrack など、さまざまな視覚タスクにわたって、幅広い代表的な DETR メソッドを一貫して改善できることを示しています。コードは https://github.com/HDETR で入手できます。
One-to-one set matching is a key design for DETR to establish its end-to-end capability, so that object detection does not require a hand-crafted NMS (non-maximum suppression) to remove duplicate detections. This end-to-end signature is important for the versatility of DETR, and it has been generalized to broader vision tasks. However, we note that there are few queries assigned as positive samples and the one-to-one set matching significantly reduces the training efficacy of positive samples. We propose a simple yet effective method based on a hybrid matching scheme that combines the original one-to-one matching branch with an auxiliary one-to-many matching branch during training. Our hybrid strategy has been shown to significantly improve accuracy. In inference, only the original one-to-one match branch is used, thus maintaining the end-to-end merit and the same inference efficiency of DETR. The method is named H-DETR, and it shows that a wide range of representative DETR methods can be consistently improved across a wide range of visual tasks, including DeformableDETR, PETRv2, PETR, and TransTrack, among others. The code is available at: https://github.com/HDETR
updated: Mon Mar 27 2023 10:27:04 GMT+0000 (UTC)
published: Tue Jul 26 2022 17:52:14 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アソシエイト