arXiv reaDer
平均平均精度の計算の並列実装
A Parallel Implementation of Computing Mean Average Precision
平均平均精度(mAP)は、オブジェクト検出器の品質を評価するために広く使用されていますが、効率的な実装はまだありません。現在の実装では、クラスのすべての検出を順番にループすることにより、一度に1つのクラスの真陽性(TP)と偽陽性(FP)のみをカウントできます。これらのアプローチは非効率的であるだけでなく、トレーニング中に検証mAPを報告するのにも不便です。推論が終了した後、mAPを即座に計算できるように、検出されたバウンディングボックス(DTBB)とグラウンドトゥルースバウンディングボックス(GTBB)のミニバッチを処理できる並列化された代替案を提案します。シーケンシャル実装のループと制御ステートメントは、ブロードキャスト、マスキング、およびインデックス作成の広範な使用に置き換えられます。関係するすべてのオペレーターは、PyTorchやTensorFlowなどの一般的な機械学習フレームワークによってサポートされています。その結果、私たちの実装ははるかに高速になり、一般的なトレーニングルーチンに簡単に適合できます。実装のPyTorchバージョンは、https://github.com/bwangca/fast-mapで入手できます。
Mean Average Precision (mAP) has been widely used for evaluating the quality of object detectors, but an efficient implementation is still absent. Current implementations can only count true positives (TP's) and false positives (FP's) for one class at a time by looping through every detection of that class sequentially. Not only are these approaches inefficient, but they are also inconvenient for reporting validation mAP during training. We propose a parallelized alternative that can process mini-batches of detected bounding boxes (DTBB's) and ground truth bounding boxes (GTBB's) as inference goes such that mAP can be instantly calculated after inference is finished. Loops and control statements in sequential implementations are replaced with extensive uses of broadcasting, masking, and indexing. All operators involved are supported by popular machine learning frameworks such as PyTorch and TensorFlow. As a result, our implementation is much faster and can easily fit into typical training routines. A PyTorch version of our implementation is available at https://github.com/bwangca/fast-map.
updated: Sun Jun 19 2022 23:23:52 GMT+0000 (UTC)
published: Sun Jun 19 2022 23:23:52 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アソシエイト