arXiv reaDer
信頼性伝播クラスター:オブジェクト検出器の可能性を最大限に引き出す
Confidence Propagation Cluster: Unleash Full Potential of Object Detectors
ほとんどのオブジェクト検出方法が、非最大抑制(NMS)と、Soft-NMSなどの改良バージョンを使用して冗長な境界ボックスを削除することによってオブジェクトを取得することは長い歴史があります。これらのNMSベースの方法に3つの側面から挑戦します。1)信頼値が最も高いバウンディングボックスは、グラウンドトゥルースボックスとのオーバーラップが最大の真のポジティブではない可能性があります。 2)冗長ボックスには抑制が必要であるだけでなく、それらの真のポジティブには信頼性の向上も必要です。 3)完全な並列処理を実現できるように、信頼値で候補ボックスを並べ替える必要はありません。この論文では、信念伝搬(BP)に触発されて、完全に並列化可能で精度が高いNMSベースの方法に代わる信頼伝搬クラスター(CP-Cluster)を提案します。 CP-Clusterでは、メッセージパッシングメカニズムをBPから借用して、冗長ボックスにペナルティを課し、収束するまで反復的に真陽性を同時に強化します。 FasterRCNN、SSD、FCOS、YOLOv3、YOLOv5、Centernetなどのさまざまな主流の検出器にCP-Clusterを適用することで、CP-Clusterの有効性を検証しました。MSCOCOでの実験では、検出器を再トレーニングせずにプラグアンドプレイ方式を使用できることが示されています。 NMSベースの方法と比較した場合、これらすべての最先端モデルの平均mAPを、それぞれ0.2から1.9の明確なマージンで改善します。ソースコードはhttps://github.com/shenyi0220/CP-Clusterで入手できます。
It has been a long history that most object detection methods obtain objects by using the non-maximum suppression (NMS) and its improved versions like Soft-NMS to remove redundant bounding boxes. We challenge those NMS-based methods from three aspects: 1) The bounding box with highest confidence value may not be the true positive having the biggest overlap with the ground-truth box. 2) Not only suppression is required for redundant boxes, but also confidence enhancement is needed for those true positives. 3) Sorting candidate boxes by confidence values is not necessary so that full parallelism is achievable. In this paper, inspired by belief propagation (BP), we propose the Confidence Propagation Cluster (CP-Cluster) to replace NMS-based methods, which is fully parallelizable as well as better in accuracy. In CP-Cluster, we borrow the message passing mechanism from BP to penalize redundant boxes and enhance true positives simultaneously in an iterative way until convergence. We verified the effectiveness of CP-Cluster by applying it to various mainstream detectors such as FasterRCNN, SSD, FCOS, YOLOv3, YOLOv5, Centernet etc. Experiments on MS COCO show that our plug and play method, without retraining detectors, is able to steadily improve average mAP of all those state-of-the-art models with a clear margin from 0.2 to 1.9 respectively when compared with NMS-based methods. Source code is available at https://github.com/shenyi0220/CP-Cluster
updated: Wed Dec 01 2021 08:22:00 GMT+0000 (UTC)
published: Wed Dec 01 2021 08:22:00 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アソシエイト