arXiv reaDer
予測されたIoUとアンカーIoUを組み合わせることによるオブジェクト検出のための動的ラベル割り当て
Dynamic Label Assignment for Object Detection by Combining Predicted IoUs and Anchor IoUs
ラベルの割り当ては、最新のオブジェクト検出モデルで重要な役割を果たします。検出モデルは、異なるラベル割り当て戦略でまったく異なるパフォーマンスをもたらす可能性があります。アンカーベースの検出モデルの場合、正のサンプルと負のサンプルがIoUしきい値で除算されるため、アンカーとそれに対応するグラウンドトゥルースバウンディングボックスの間のIoU(Intersection over Union)しきい値が重要な要素になります。初期のオブジェクト検出器は、すべてのトレーニングサンプルに固定しきい値を使用するだけですが、最近の検出アルゴリズムは、グラウンドトゥルースボックスへのIoUの分布に基づく適応しきい値に焦点を合わせています。この論文では、予測を伴うトレーニングステータスに基づいて動的にラベル割り当てを実行するためのシンプルで効果的なアプローチを紹介します。ラベル割り当てに予測を導入することにより、グラウンドトゥルースオブジェクトに対してより高いIoUを持つより高品質のサンプルがポジティブサンプルとして選択されます。これにより、分類スコアとIoUスコアの間の不一致が減少し、より高品質の境界ボックスが生成されます。 。私たちのアプローチは、適応ラベル割り当てアルゴリズムとそれらのポジティブサンプルのバウンディングボックス損失による検出モデルのパフォーマンスの改善を示しており、より高品質の予測ボックスを持つより多くのサンプルがポジティブとして選択されることを示しています。
Label assignment plays a significant role in modern object detection models. Detection models may yield totally different performances with different label assignment strategies. For anchor-based detection models, the IoU (Intersection over Union) threshold between the anchors and their corresponding ground truth bounding boxes is the key element since the positive samples and negative samples are divided by the IoU threshold. Early object detectors simply utilize the fixed threshold for all training samples, while recent detection algorithms focus on adaptive thresholds based on the distribution of the IoUs to the ground truth boxes. In this paper, we introduce a simple while effective approach to perform label assignment dynamically based on the training status with predictions. By introducing the predictions in label assignment, more high-quality samples with higher IoUs to the ground truth objects are selected as the positive samples, which could reduce the discrepancy between the classification scores and the IoU scores, and generate more high-quality boundary boxes. Our approach shows improvements in the performance of the detection models with the adaptive label assignment algorithm and lower bounding box losses for those positive samples, indicating more samples with higher-quality predicted boxes are selected as positives.
updated: Mon Jul 11 2022 23:42:10 GMT+0000 (UTC)
published: Sun Jan 23 2022 23:14:07 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アソシエイト