変圧器によるエンドツーエンドオブジェクト検出
- ソフト名 : DETR
- OS :
- 言語 : Python
- 開発者 : HOME
DETR 詳細説明
DETR(検出トランス)のためのPYTORCHトレーニングコードと登録されたモデルフル複雑な手作りのオブジェクト検出パイプラインをトランスフォーマーに置き換え、R-CNNをより速いR-CNNと交換し、計算パワーの半分を使用してCOCO上の42 APを取得し、同じ数のパラメータを使用します。Pytorchの50行の推論。それが何であるか。従来のコンピュータビジョン技術とは異なり、DETRはオブジェクト検出に直接設定予測問題として近づきます。それはセットベースのグローバルロスで構成されており、これはバイパリットマッチングを介して固有の予測とトランスエンコーダデコーダアーキテクチャを強制します。修正された学習オブジェクトクエリの固定された小さなセットを考えると、最終的な予測セットを並行して直接出力するためのオブジェクトとグローバルイメージコンテキストの関係に関する理由があります。この平行な性質のために、DETRは非常に速く効率的です。