CycleGAN_风格迁移+qt界面

直接上效果图

CycleGAN_风格迁移+qt界面_第1张图片

 代码文件展示

CycleGAN_风格迁移+qt界面_第2张图片

其中data文件夹下是数据集文件,这里是黄马文件夹和斑马文件夹,运行01train_cyclegan.py进行训练,得到的模型存在weights文件夹下。

运行02pyqt界面.py是可视化界面可以选择要风格变换的图片,调用训练好的模型进行识别。

你可能感兴趣的:(python,开发语言)