【08/04更新】在前几天的Commit中,Contextual Loss已经支持多GPU训练
1.Background
对于图像风格迁移,最常用的做法就是通过GAN网络实现,然而,如果你没有很强大的算力,训练GAN网络就是比较困难的一件事儿了。基于传统的前向CNN网络,有相关学者提出了一些感知loss,使用该loss也可以完成图像风格迁移,比较常见的就是Contextual Loss。
首先给出论文作者开源出的源码,基于Tensorflow实现
roimehrez/contextualLosswww.github.com代码中包含了风格迁移的几个样例,但我没去测试。我的重点在于将Contextual Loss应用于其他图像增强的领域,来尽可能多地保留图像细节,下面是我用PyTorch对Contextual Loss的实现,现已支持多GPU运行~
z-bingo/Contextual-Loss-PyTorchgithub.com