arXiv reaDer
ツリーのエネルギー損失:まばらに注釈が付けられたセマンティックセグメンテーションに向けて
Tree Energy Loss: Towards Sparsely Annotated Semantic Segmentation
スパースアノテーション付きセグメンテーションセグメンテーション(SASS)は、粗粒度(つまり、ポイント、スクリブル、およびブロックワイズ)の監視を使用してセグメンテーションネットワークをトレーニングすることを目的としています。各画像では、ピクセルのごく一部のみがラベル付けされます。この論文では、ラベル付けされていないピクセルのセマンティックガイダンスを提供することにより、SASSの新しいツリーエネルギー損失を提案します。ツリーエネルギー損失は、低レベルと高レベルの両方のペアワイズアフィニティをモデル化するための最小スパニングツリーとして画像を表します。これらのアフィニティをネットワーク予測に順次適用することにより、ラベルのないピクセルのソフト疑似ラベルが粗い方法から細かい方法で生成され、動的なオンラインセルフトレーニングが実現されます。ツリーのエネルギー損失は効果的であり、従来のセグメンテーション損失と組み合わせることで既存のフレームワークに簡単に組み込むことができます。以前のSASSの方法と比較して、私たちの方法は、すべてのSASS設定でそれらを上回りながら、多段階のトレーニング戦略、交互の最適化手順、追加の監視ありデータ、または時間のかかる後処理を必要としません。コードはhttps://github.com/megvii-research/TreeEnergyLossで入手できます。
Sparsely annotated semantic segmentation (SASS) aims to train a segmentation network with coarse-grained (i.e., point-, scribble-, and block-wise) supervisions, where only a small proportion of pixels are labeled in each image. In this paper, we propose a novel tree energy loss for SASS by providing semantic guidance for unlabeled pixels. The tree energy loss represents images as minimum spanning trees to model both low-level and high-level pair-wise affinities. By sequentially applying these affinities to the network prediction, soft pseudo labels for unlabeled pixels are generated in a coarse-to-fine manner, achieving dynamic online self-training. The tree energy loss is effective and easy to be incorporated into existing frameworks by combining it with a traditional segmentation loss. Compared with previous SASS methods, our method requires no multistage training strategies, alternating optimization procedures, additional supervised data, or time-consuming post-processing while outperforming them in all SASS settings. Code is available at https://github.com/megvii-research/TreeEnergyLoss.
updated: Tue Mar 22 2022 04:17:36 GMT+0000 (UTC)
published: Mon Mar 21 2022 05:16:23 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アソシエイト