pythonでSoft-NMS(Soft Non-Maximum Suppression)を実装する方法を紹介します。 Soft-NMSは、SSDやYOLOといった物体検出AIの後処理で使用されるNMSの改良型アルゴリズムです。 この記事でできること ニューラルネットワーク(ディープラーニング)等を用いた機械学習による物体検出AIでは、 たくさんの検出枠(この記事では矩形と呼んでいます)が出力されます。 従来はNMSという処理で、重複する矩形の統合、削除処理を行いますが、この記事ではNMSの改良型であるSoft-NMSの実装例を紹介します。 次の画像は、いらすとやの「離れて給食を食べる子供のイラスト」の二人の子供の顔付近にいくつかの矩形を付与した画像です。 「離れて給食を食べる子供のイラスト」に矩形を付与した画像 この画像にSoft-NMSを適用した例が次の画像です。 Soft-NMSを適