arXiv reaDer
ISTR:トランスフォーマーを使用したエンドツーエンドのインスタンスセグメンテーション
ISTR: End-to-End Instance Segmentation with Transformers
エンドツーエンドのパラダイムは、さまざまな深層学習ベースのコンピュータービジョンモデルの精度を大幅に向上させます。この目的のために、オブジェクト検出などのタスクは、トレーニングによる非最大抑制の削除など、非エンドツーエンドコンポーネントを、2部マッチングに基づくセット損失で置き換えることによってアップグレードされました。ただし、このようなアップグレードは、オブジェクト検出と比較して出力ディメンションが大幅に高いため、インスタンスのセグメンテーションには適用できません。この論文では、この種の最初のエンドツーエンドフレームワークであるISTRと呼ばれるインスタンスセグメンテーショントランスフォーマーを提案します。 ISTRは、低次元のマスク埋め込みを予測し、それらをセット損失のグラウンドトゥルースマスク埋め込みと照合します。さらに、ISTRは、検出とセグメンテーションを繰り返しの改良戦略と同時に実行します。これにより、既存のトップダウンおよびボトムアップフレームワークと比較して、インスタンスのセグメンテーションを実現する新しい方法が提供されます。提案されたエンドツーエンドメカニズムの恩恵を受けて、ISTRは、近似ベースの次善の埋め込みでも最先端のパフォーマンスを発揮します。具体的には、ISTRは、MSCOCOデータセットでResNet50-FPNを使用して46.8 / 38.6ボックス/マスクAPを取得し、ResNet101-FPNを使用して48.1 /39.9ボックス/マスクAPを取得します。定量的および定性的な結果は、インスタンスレベルの認識のための確かなベースラインとしてのISTRの有望な可能性を明らかにしています。コードはhttps://github.com/hujiecpp/ISTRで入手できます。
End-to-end paradigms significantly improve the accuracy of various deep-learning-based computer vision models. To this end, tasks like object detection have been upgraded by replacing non-end-to-end components, such as removing non-maximum suppression by training with a set loss based on bipartite matching. However, such an upgrade is not applicable to instance segmentation, due to its significantly higher output dimensions compared to object detection. In this paper, we propose an instance segmentation Transformer, termed ISTR, which is the first end-to-end framework of its kind. ISTR predicts low-dimensional mask embeddings, and matches them with ground truth mask embeddings for the set loss. Besides, ISTR concurrently conducts detection and segmentation with a recurrent refinement strategy, which provides a new way to achieve instance segmentation compared to the existing top-down and bottom-up frameworks. Benefiting from the proposed end-to-end mechanism, ISTR demonstrates state-of-the-art performance even with approximation-based suboptimal embeddings. Specifically, ISTR obtains a 46.8/38.6 box/mask AP using ResNet50-FPN, and a 48.1/39.9 box/mask AP using ResNet101-FPN, on the MS COCO dataset. Quantitative and qualitative results reveal the promising potential of ISTR as a solid baseline for instance-level recognition. Code has been made available at: https://github.com/hujiecpp/ISTR.
updated: Thu May 06 2021 03:10:33 GMT+0000 (UTC)
published: Mon May 03 2021 06:00:09 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アソシエイト