arXiv reaDer
MPDIoU: 効率的かつ正確なバウンディングボックス回帰の損失
MPDIoU: A Loss for Efficient and Accurate Bounding Box Regression
バウンディング ボックス回帰 (BBR) は、オブジェクトの位置特定における重要なステップであるオブジェクトの検出とインスタンスのセグメンテーションで広く使用されています。ただし、予測ボックスのアスペクト比がグラウンドトゥルース ボックスと同じであるが、幅と高さの値がまったく異なる場合、バウンディング ボックス回帰用の既存の損失関数のほとんどは最適化できません。上記の問題に取り組むために、水平方向の長方形の幾何学的特徴を徹底的に調査し、最小点距離に基づく新しい境界ボックス類似性比較メトリック MPDIoU を提案します。これには、既存の損失関数で考慮されたすべての関連要素が含まれています。重なり合わない領域、中心点の距離、幅と高さの偏差を計算しながら、計算プロセスを簡素化します。これに基づいて、MPDIoU に基づく LMPDIoU と呼ばれる境界ボックス回帰損失関数を提案します。実験結果は、MPDIoU 損失関数が最先端のインスタンス セグメンテーション (例: YOLACT) に適用され、PASCAL VOC、MS COCO、および IIIT5k でトレーニングされたオブジェクト検出 (例: YOLOv7) モデルが既存の損失関数よりも優れていることを示しています。
Bounding box regression (BBR) has been widely used in object detection and instance segmentation, which is an important step in object localization. However, most of the existing loss functions for bounding box regression cannot be optimized when the predicted box has the same aspect ratio as the groundtruth box, but the width and height values are exactly different. In order to tackle the issues mentioned above, we fully explore the geometric features of horizontal rectangle and propose a novel bounding box similarity comparison metric MPDIoU based on minimum point distance, which contains all of the relevant factors considered in the existing loss functions, namely overlapping or non-overlapping area, central points distance, and deviation of width and height, while simplifying the calculation process. On this basis, we propose a bounding box regression loss function based on MPDIoU, called LMPDIoU . Experimental results show that the MPDIoU loss function is applied to state-of-the-art instance segmentation (e.g., YOLACT) and object detection (e.g., YOLOv7) model trained on PASCAL VOC, MS COCO, and IIIT5k outperforms existing loss functions.
updated: Fri Jul 14 2023 23:54:49 GMT+0000 (UTC)
published: Fri Jul 14 2023 23:54:49 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アソシエイト