arXiv reaDer
セグメンテーション混乱による生成画像修復 敵対的トレーニングと対照的学習
Generative Image Inpainting with Segmentation Confusion Adversarial Training and Contrastive Learning
この論文では、セグメンテーション混乱敵対的トレーニング (SCAT) と対照学習を使用した画像修復のための新しい敵対的トレーニング フレームワークを紹介します。 SCAT は、修復ジェネレーターとセグメンテーション ネットワークの間で敵対的なゲームを行います。セグメンテーション ネットワークは、ピクセル レベルのローカル トレーニング信号を提供し、自由形式の穴のある画像に適応できます。 SCAT を標準的なグローバルな敵対的トレーニングと組み合わせることにより、新しい敵対的トレーニング フレームワークは、(1) 修復された画像のグローバルな一貫性、(2) 修復された画像の局所的な細かいテクスチャの詳細、および (3)自由形式の穴のある画像を柔軟に処理できます。さらに、弁別器の特徴表現空間を活用することにより、修復モデルのトレーニングを安定化および改善するために、テクスチャおよびセマンティックの対照的な学習損失を提案します。この空間では、修復画像はグラウンド トゥルース画像に近づきますが、破損した画像からは遠ざけられます。提案されたコントラスト損失は、修復された画像を、特徴表現空間内の破損した画像データ ポイントから実際の画像データ ポイントに移動するように誘導し、よりリアルな完成画像をもたらします。 2 つのベンチマーク データセットで大規模な実験を行い、モデルの有効性と優位性を質的および量的に実証します。
This paper presents a new adversarial training framework for image inpainting with segmentation confusion adversarial training (SCAT) and contrastive learning. SCAT plays an adversarial game between an inpainting generator and a segmentation network, which provides pixel-level local training signals and can adapt to images with free-form holes. By combining SCAT with standard global adversarial training, the new adversarial training framework exhibits the following three advantages simultaneously: (1) the global consistency of the repaired image, (2) the local fine texture details of the repaired image, and (3) the flexibility of handling images with free-form holes. Moreover, we propose the textural and semantic contrastive learning losses to stabilize and improve our inpainting model's training by exploiting the feature representation space of the discriminator, in which the inpainting images are pulled closer to the ground truth images but pushed farther from the corrupted images. The proposed contrastive losses better guide the repaired images to move from the corrupted image data points to the real image data points in the feature representation space, resulting in more realistic completed images. We conduct extensive experiments on two benchmark datasets, demonstrating our model's effectiveness and superiority both qualitatively and quantitatively.
updated: Thu Mar 23 2023 09:34:17 GMT+0000 (UTC)
published: Thu Mar 23 2023 09:34:17 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アソシエイト