Saved in:
Bibliographic Details
Main Authors: Zhao, Chen, Cai, Wei-Ling, Yuan, Zheng
Format: Preprint
Published: 2023
Subjects:
Online Access:https://arxiv.org/abs/2304.11319
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911792009052160
author Zhao, Chen
Cai, Wei-Ling
Yuan, Zheng
author_facet Zhao, Chen
Cai, Wei-Ling
Yuan, Zheng
contents Existing image-to-image (I2I) translation methods achieve state-of-the-art performance by incorporating the patch-wise contrastive learning into Generative Adversarial Networks. However, patch-wise contrastive learning only focuses on the local content similarity but neglects the global structure constraint, which affects the quality of the generated images. In this paper, we propose a new unpaired I2I translation framework based on dual contrastive regularization and spectral normalization, namely SN-DCR. To maintain consistency of the global structure and texture, we design the dual contrastive regularization using different deep feature spaces respectively. In order to improve the global structure information of the generated images, we formulate a semantic contrastive loss to make the global semantic structure of the generated images similar to the real images from the target domain in the semantic feature space. We use Gram Matrices to extract the style of texture from images. Similarly, we design a style contrastive loss to improve the global texture information of the generated images. Moreover, to enhance the stability of the model, we employ the spectral normalized convolutional network in the design of our generator. We conduct comprehensive experiments to evaluate the effectiveness of SN-DCR, and the results prove that our method achieves SOTA in multiple tasks.
format Preprint
id arxiv_https___arxiv_org_abs_2304_11319
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Spectral Normalization and Dual Contrastive Regularization for Image-to-Image Translation
Zhao, Chen
Cai, Wei-Ling
Yuan, Zheng
Computer Vision and Pattern Recognition
Existing image-to-image (I2I) translation methods achieve state-of-the-art performance by incorporating the patch-wise contrastive learning into Generative Adversarial Networks. However, patch-wise contrastive learning only focuses on the local content similarity but neglects the global structure constraint, which affects the quality of the generated images. In this paper, we propose a new unpaired I2I translation framework based on dual contrastive regularization and spectral normalization, namely SN-DCR. To maintain consistency of the global structure and texture, we design the dual contrastive regularization using different deep feature spaces respectively. In order to improve the global structure information of the generated images, we formulate a semantic contrastive loss to make the global semantic structure of the generated images similar to the real images from the target domain in the semantic feature space. We use Gram Matrices to extract the style of texture from images. Similarly, we design a style contrastive loss to improve the global texture information of the generated images. Moreover, to enhance the stability of the model, we employ the spectral normalized convolutional network in the design of our generator. We conduct comprehensive experiments to evaluate the effectiveness of SN-DCR, and the results prove that our method achieves SOTA in multiple tasks.
title Spectral Normalization and Dual Contrastive Regularization for Image-to-Image Translation
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2304.11319