arXiv reaDer
RGB-D顕著なオブジェクト検出のための自己監視事前トレーニング
Self-Supervised Pretraining for RGB-D Salient Object Detection
既存のCNNベースのRGB-D顕著なオブジェクト検出(SOD)ネットワークはすべて、適切な初期化を提供するのに役立つ階層機能を学習するために、ImageNetで事前トレーニングする必要があります。ただし、大規模なデータセットの収集と注釈付けには、時間と費用がかかります。この論文では、自己監視表現学習(SSL)を利用して、2つの口実タスクを設計します。クロスモーダルオートエンコーダと深度輪郭推定です。私たちの口実タスクは、事前トレーニングを実行するために少数のラベルのないRGB-Dデータセットのみを必要とします。これにより、ネットワークは豊富なセマンティックコンテキストをキャプチャし、2つのモダリティ間のギャップを減らし、ダウンストリームタスクの効果的な初期化を提供します。さらに、RGB-D SODのクロスモーダル融合に固有の問題について、単一の機能融合をマルチパス融合に分割して、一貫性のある差分情報の適切な認識を実現する一貫性差分集約(CDA)モジュールを提案します。 。 CDAモジュールは一般的であり、クロスモーダルおよびクロスレベルの機能融合に適しています。 6つのベンチマークデータセットでの広範な実験は、自己監視の事前トレーニング済みモデルが、ImageNetで事前トレーニングされたほとんどの最先端の方法に対して良好に機能することを示しています。ソースコードはredhttps://github.com/Xiaoqi-Zhao-DLUT/SSLSODで公開されます。
Existing CNNs-Based RGB-D salient object detection (SOD) networks are all required to be pretrained on the ImageNet to learn the hierarchy features which helps provide a good initialization. However, the collection and annotation of large-scale datasets are time-consuming and expensive. In this paper, we utilize self-supervised representation learning (SSL) to design two pretext tasks: the cross-modal auto-encoder and the depth-contour estimation. Our pretext tasks require only a few and unlabeled RGB-D datasets to perform pretraining, which makes the network capture rich semantic contexts and reduce the gap between two modalities, thereby providing an effective initialization for the downstream task. In addition, for the inherent problem of cross-modal fusion in RGB-D SOD, we propose a consistency-difference aggregation (CDA) module that splits a single feature fusion into multi-path fusion to achieve an adequate perception of consistent and differential information. The CDA module is general and suitable for cross-modal and cross-level feature fusion. Extensive experiments on six benchmark datasets show that our self-supervised pretrained model performs favorably against most state-of-the-art methods pretrained on ImageNet. The source code will be publicly available at redhttps://github.com/Xiaoqi-Zhao-DLUT/SSLSOD.
updated: Fri Dec 03 2021 13:17:30 GMT+0000 (UTC)
published: Fri Jan 29 2021 09:16:06 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アソシエイト