21,000クラスを検出可能な物体検出手法 Detic
Detecting Twenty-thousand Classes using Image-level Supervision
- https://arxiv.org/abs/2201.02605
- 2022/01
- faster RCNN ベースの2段階モデルで候補領域提案後に分類部分を訓練する方法で,imagenetの21,000クラスの物体検出が可能なDetector with image classes (Detic)を提案
- 従来の弱教師検出手法では領域提案技術で出力されるboxとクラスラベルのみを持つ画像分類データセットのラベルをマッチさせる戦略を取っていたが,本論文は領域提案だけで未知の物体の提案は学習できているとしている
- boxラベルのない画像分類データに対しては,モデル出力候補のうちエリアが最大の出力に対してクラスラベルを使って分類を学習.lossは全クラスに対してBCE.ここではlocalization損失は計算しない
- boxラベルがあるデータについては通常の物体検出学習
- 分類部分はCLIPを使用しており,open vocabularyな分類が可能
実験ではfrequent, common, rareの3グループを持つ物体検出データセットのLVISと,画像分類用データセットとして21Kクラスのimagenet,conceptual captionsを使用.conceptual captionはimage captionデータセットなのでキャプションからクラスラベルを抽出してLVISクラスに言及しているものを使用
https://huggingface.co/spaces/akhaliq/Detic
- デモ