contrastive unpaired translation:对比性的非配对图像到图像翻译比cyclegan更快更轻松的训练(ECCV 2020在PyTor

对比非成对翻译(CUT) | | | 我们提供了基于补丁式对比学习和对抗学习的不成对图像到图像翻译的PyTorch实现。 没有使用手工制作的损耗和逆网络。 与相比,我们的模型训练更快,内存占用更少。 另外,我们的方法可以扩展到单个图像训练,其中每个“域”仅是单个图像。 , ,,加州大学伯克利分校和Adobe Research 在ECCV 2020中 伪代码 import torch cross_entropy_loss = torch . nn . CrossEntropyLoss () # Input: f_q (BxCxS) and sampled features from H(G_enc(x)) # Input: f_k (BxCxS) are sampled features from H(G_enc(G(x)) # Input: tau is the temperatu