偶然看到有个b站视频,是关于如何实现一个图片的赛博朋克风,通过调整色调就可以实现。然后就看到有博主使用python的opencv库实现这个效果。
因此,我想使用Cyclegan生成对抗网络实现风格迁移。
先是准备数据集,我在https://wallhaven.cc/网站爬取了了大约2000张赛博朋克风的图片,因为设备条件有限,图片都是320大小的:
另外我在该网站爬取了了现实城市和风景的图片大概两千多张:
利用论文作者提供的github代码,并修改训练参数、准备数据集,进行训练:(训练的中间结果)
因为这个网络分辨率比较低,所以效果一般,此外,数据集也有一些问题,中间有些脏数据。有些图片并不是赛博朋克风格。