项目简介
本项目借助神经网络达成图像风格迁移功能,可把一张图片的风格迁移至另一张图片。主要运用Python语言与TensorFlow框架,通过预训练的VGG19模型开展特征提取,进而实现风格迁移。
项目的主要特性和功能
- 运用预训练的VGG19模型提取图像特征,为风格迁移奠定基础。
- 具备风格迁移功能,能将内容图片的风格迁移到样式图片上,生成新图像。
- 支持用户自定义训练周期数、每周期训练次数等参数。
- 在训练过程中生成并保存风格迁移后的图片,方便用户直观查看效果。
安装使用步骤
- 复制项目并安装依赖
使用以下命令安装项目所需的依赖环境:
bash pip install -r requirements.txt
- 修改配置参数
打开
settings.py
文件,按需修改训练周期数、每周期训练次数、图片路径等参数。 - 运行训练脚本
使用以下命令开始训练:
bash python train.py
- 查看生成图片
训练完成后,生成的图片将保存在默认的输出目录(
./output
)中,用户可查看风格迁移的效果。
下载地址
点击下载 【提取码: 4003】【解压密码: www.makuang.net】