arXiv reaDer
距離IoUロス: バウンディングボックス回帰のためのより高速でより良い学習
Distance-IoU Loss: Faster and Better Learning for Bounding Box Regression
バウンディングボックス回帰は、物体検出の重要なステップである。既存の手法では、バウンディングボックス回帰ではℓ_n-normロスが広く採用されているが、評価指標であるIoU (Intersection over Union)には対応していない。近年、IoUロスや一般化IoU (GIoU)ロスが提案されているが、収束が遅く、回帰が不正確であるという問題がある。本論文では、予測されたボックスとターゲットボックス間の正規化された距離を組み込んだ距離IoU (Distance-IoU)ロスを提案し、IoUロスやGIoUロスよりもはるかに高速に収束する。さらに、本論文では、バウンディングボックス回帰におけるオーバーラップ面積、中心点距離、アスペクト比の3つの幾何学的要因をまとめ、これに基づいて完全IoU (CIoU)ロスを提案することで、より高速な収束と性能向上を実現した。YOLO v3、SSD、Faster RCNNなどの最新の物体検出アルゴリズムにDIoUとCIoU損失を組み込むことで、IoU指標だけでなくGIoU指標の面でも顕著な性能向上を実現している。また、DIoUはNMS (Non-maximum suppression)に容易に採用でき、基準として機能し、性能向上をさらに促進することができる。ソースコードと学習モデルは に掲載されている。
Bounding box regression is the crucial step in object detection. In existing methods, while ℓ_n-norm loss is widely adopted for bounding box regression, it is not tailored to the evaluation metric, i.e., Intersection over Union (IoU). Recently, IoU loss and generalized IoU (GIoU) loss have been proposed to benefit the IoU metric, but still suffer from the problems of slow convergence and inaccurate regression. In this paper, we propose a Distance-IoU (DIoU) loss by incorporating the normalized distance between the predicted box and the target box, which converges much faster in training than IoU and GIoU losses. Furthermore, this paper summarizes three geometric factors in bounding box regression, i.e. , overlap area, central point distance and aspect ratio, based on which a Complete IoU (CIoU) loss is proposed, thereby leading to faster convergence and better performance. By incorporating DIoU and CIoU losses into state-of-the-art object detection algorithms, e.g., YOLO v3, SSD and Faster RCNN, we achieve notable performance gains in terms of not only IoU metric but also GIoU metric. Moreover, DIoU can be easily adopted into non-maximum suppression (NMS) to act as the criterion, further boosting performance improvement. The source code and trained models are available at
updated: Tue Nov 19 2019 14:20:07 GMT+0000 (UTC)
published: Tue Nov 19 2019 14:20:07 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)アソシエイト