Csdn上上传的代码都是完整的,可以看我对应的博客都是有效果图的。也可以看演示视频,防治踩坑。
需要查看其他代码的可看b站视频演示:
https://space.bilibili.com/124080712?spm_id_from=333.1007.0.0
其他小项目完整代码:
https://blog.csdn.net/qq_34904125?type=download
算法部分
本次是油画风格迁移,数据集分别是真实图片和油画图片,
分别放在data文件夹的white和color文件夹下。
代码属于傻瓜式简单操作,简单易上手
代码依次运行
python 01train_cyclegan.py
会训练模型,并保存在weights文件夹下。
训练epoch越多效果越好,
训练的一些参数设置在01demo的开头,可以自行修改。
python 02detector_simple.py
调用训练好的模型对图片进行风格迁移,在代码的最后一行修改图片路径即可对其他图片进行识别。