arXiv reaDer
SOTR:トランスフォーマーを使用したオブジェクトのセグメント化
SOTR: Segmenting Objects with Transformers
最新のトランスベースモデルは、畳み込みニューラルネットワーク(CNN)よりも優れた、ビジョンタスクでの優れたパフォーマンスを示しています。この作業では、高品質のインスタンスセグメンテーションのための斬新で柔軟かつ効果的なトランスベースのモデルを紹介します。提案された方法であるTRansformersを使用したオブジェクトのセグメント化(SOTR)は、セグメンテーションパイプラインを簡素化し、2つの並列サブタスクが追加された代替CNNバックボーン上に構築します:(1)トランスフォーマーを介してインスタンスごとのカテゴリを予測し、(2)マルチでセグメンテーションマスクを動的に生成します-レベルのアップサンプリングモジュール。 SOTRは、機能ピラミッドネットワーク(FPN)とツイントランスフォーマーによって、それぞれ低レベルの機能表現を効果的に抽出し、長距離のコンテキスト依存関係をキャプチャできます。一方、元のトランスフォーマーと比較して、提案されたツイントランスフォーマーは、ピクセルのエンコードに行と列の注意のみが含まれるため、時間とリソースの効率が高くなります。さらに、SOTRは、さまざまなCNNバックボーンおよびトランスモデルバリアントに簡単に組み込むことができ、セグメンテーションの精度とトレーニングの収束を大幅に改善します。広範な実験により、SOTRはMS COCOデータセットで良好に機能し、最先端のインスタンスセグメンテーションアプローチを上回っています。シンプルでありながら強力なフレームワークが、インスタンスレベルの認識の優先ベースラインとして役立つことを願っています。私たちのコードはhttps://github.com/easton-cau/SOTRで入手できます。
Most recent transformer-based models show impressive performance on vision tasks, even better than Convolution Neural Networks (CNN). In this work, we present a novel, flexible, and effective transformer-based model for high-quality instance segmentation. The proposed method, Segmenting Objects with TRansformers (SOTR), simplifies the segmentation pipeline, building on an alternative CNN backbone appended with two parallel subtasks: (1) predicting per-instance category via transformer and (2) dynamically generating segmentation mask with the multi-level upsampling module. SOTR can effectively extract lower-level feature representations and capture long-range context dependencies by Feature Pyramid Network (FPN) and twin transformer, respectively. Meanwhile, compared with the original transformer, the proposed twin transformer is time- and resource-efficient since only a row and a column attention are involved to encode pixels. Moreover, SOTR is easy to be incorporated with various CNN backbones and transformer model variants to make considerable improvements for the segmentation accuracy and training convergence. Extensive experiments show that our SOTR performs well on the MS COCO dataset and surpasses state-of-the-art instance segmentation approaches. We hope our simple but strong framework could serve as a preferment baseline for instance-level recognition. Our code is available at https://github.com/easton-cau/SOTR.
updated: Sun Aug 15 2021 14:10:11 GMT+0000 (UTC)
published: Sun Aug 15 2021 14:10:11 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アソシエイト