arXiv reaDer
ガウスワッサースタイン距離損失による回転物体検出の再考
Rethinking Rotated Object Detection with Gaussian Wasserstein Distance Loss
境界の不連続性と最終的な検出メトリックとの不一致が、回転検出回帰損失設計のボトルネックになっています。本論文では、問題を解決するための基本的なアプローチとして、ガウスワッサースタイン距離に基づく新しい回帰損失を提案します。具体的には、回転バウンディングボックスは2次元ガウス分布に変換されます。これにより、勾配逆伝播によって効率的に学習できるガウスワッサースタイン距離(GWD)によって、区別できない回転IoU誘導損失を概算できます。 GWDは、小さなオブジェクトの検出でよくある2つの回転する境界ボックスの間にオーバーラップがない場合でも、学習に役立つ可能性があります。 3つの独自のプロパティのおかげで、GWDは、境界ボックスの定義方法に関係なく、境界の不連続性と正方形のような問題をエレガントに解決することもできます。異なる検出器を使用した5つのデータセットでの実験は、私たちのアプローチの有効性を示しています。コードはhttps://github.com/yangxue0827/RotationDetectionで入手できます。
Boundary discontinuity and its inconsistency to the final detection metric have been the bottleneck for rotating detection regression loss design. In this paper, we propose a novel regression loss based on Gaussian Wasserstein distance as a fundamental approach to solve the problem. Specifically, the rotated bounding box is converted to a 2-D Gaussian distribution, which enables to approximate the indifferentiable rotational IoU induced loss by the Gaussian Wasserstein distance (GWD) which can be learned efficiently by gradient back-propagation. GWD can still be informative for learning even there is no overlapping between two rotating bounding boxes which is often the case for small object detection. Thanks to its three unique properties, GWD can also elegantly solve the boundary discontinuity and square-like problem regardless how the bounding box is defined. Experiments on five datasets using different detectors show the effectiveness of our approach. Codes are available at https://github.com/yangxue0827/RotationDetection.
updated: Sat May 08 2021 10:30:13 GMT+0000 (UTC)
published: Thu Jan 28 2021 12:04:35 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アソシエイト