基于卷积神经网络图片风格迁移实践

fork的项目保存在自己GitHub

GitHub - hzy46/fast-neural-style-tensorflow: A tensorflow implementation for fast neural style!

requirements:

1、安装python 2.7, tensorflow>=1.0,linux

2、ubuntu 18.0.4

一、实验目的:

1、 掌握卷积神经网络方法

2、掌握图片迁移的网络架构方法

二、实验内容

1、安装kylin版本的Ubuntu

2、训练、测试神经网络

3、根据四种图片风格,选择四张待迁移的图片,经过测试生成风格迁移后的图片

三、实验要求

1、安装python 2.7, tensorflow>=1.0,linux

2、安装好ubuntu,在论文主页下载模型框架

四、实验步骤
一、安装ubuntu和虚拟机
1、在VMwareWorkstatiion官网下载相关版本的虚拟机,并且在网上下载ubuntu18版本的光盘映像文件
 ubuntu官网:https://cn.ubuntu.com/download




2、打开虚拟机,点击创建新的虚拟机


 3、选用典型配置



 4、选择磁盘大小


 5、选择Linux操作系统,Ubuntu64位版本



 6、自定义输入虚拟机名称和安装位置




 7、选择将虚拟磁盘拆分位多个文件选项,并且选择适当的磁盘大小




8、点击编辑虚拟机配置,在CD/DVD中选择刚刚下载的ISO映像文件,点击确认按钮后点击开启此虚拟机进入



 9、选用中文(简体)后安装Ubuntu



 10、键盘布局全部选择中文布局



 11、跟新和其他软件按如图显示



 12、安装类型选择清除整个磁盘并安装Ubuntu”



 13、完善你的基本信息




 二、在ubuntu18.04)下安装Tensorflow
 1、进入ubuntu


 2、 建立一个专门的tensorlow用户用来测试本项目,用户名tensorflowLab
建立用户:sudo useradd -m tensorflowLab
给建立的用户设置密码:sudo passwd tensorflowLab


 3、进入tensorflowLab
进入su tensorflowLab

 4、在root权限下操作
对新用户增加管理员权限
sudo adduser tensorflowLab sudo
也就是开通权限。


三、安装相关环境和工具


安装依赖python-pip, python-dev,python-virtualenv
sudo apt-get install python-pip python-dev python-virtualenv
(注:这里我安装的是python 2.7.17版本)
1、初次下载时,界面图


1.1下载过,再输入相同指令,界面图

1.1.1如果不知道python版本,可以输入
python --version查找

四、创建虚拟环境virtualenv

1、创建虚拟环境,在自己目录~下创建虚拟环境变量tensorflow
virtualenv --system-site-packages ~/tensorflow

五、进入虚拟环境


 1、进入虚拟环境
cd ~/tensorflow

 2、激活环境
source bin/activate

六、在虚拟环境安装最新版本的tensorflow
 pip install  tensorflow==1.4.0

注:里我安装的是tensorflow1.4.0版本。在安装tensorflow时,选择的版本一定要和前面安装的python版本相匹配

七、工程项目与神经网络模型下载

1、接下啦下载训练好的模型model,  提前训练的。可以下载直接用(不要再重新训练),将下列链接复制到ubuntu下的firefox搜索栏,输入以下地址:

                          http://pan.baidu.com/s/1mi9Ilyo
但是进入网盘,必须事先安装linux下的网盘软件


 2、选择deb格式安装。
安装linux版的百度网盘,打开直接点击

 3、下载百度网盘后将models文件夹放在工程fast-neural-style-tensorflow-master的根目录下

 4、下载项目(图片迁移的工程project
先进入用户tensorflowLab

 4.1
打开firefox浏览器,输入Github下载地址
Github上下载项目代码地址,进入  Github地址:GitHub - hzy46/fast-neural-style-tensorflow: A tensorflow implementation for fast neural style!


 4.2
由于工程项目fast-neural-style-tensorflow-master.zip是在Downloads目录下,我希望所有项目放在用户tensorflowLab下,所以采用cp复制,输入以下复制命令就可以将项目复制到tensorflowLab
sudo cp -rp /home/Downloads/fast-neural-style-tensorflow-master.zip  /home/tensorflowLab
查找一下,有了

 注:这里我是通过提前在windows系统里把所需要的文件下载好,通过qq邮箱上传,避免了再下百度网盘,速度快


 八、 进入虚拟环境tensorflow
进入环境  :  cd ~/tensorflow

并激活  :  source bin/activate
先增加该文件的权限:

sudo chmod 777 fast-neural-style-tensorflow-master.zip

文件解压:sudo unzip fast-neural-style-tensorflow-master.zip


将训练好的模型models,复制到工程项目下面,使用以下命令完成复制神经网络训练好的模型:
sudo cp -r  /home/Downloads/models   /home/tensorflowLab/fast-neural-style-tensorflow-master

发现已经存在了!


 九、模型使用快速测试
用训练好的模型进行实现快速风格迁移

python eval.py --model_file   <your path to wave.ckpt-done>   --image_file img/test.jpg

your path to wave.ckpt-done      训练好的下载模型的路径

img/test.jpg   用于测试图片而输入测试图片的路径

切记:切换到当前工程项目的目录fast-neural-style-tensorflow-master下!

采用:python eval.py --model_file /models/wave.ckpt-done --image_file img/z1.jpg

猜你喜欢

转载自blog.csdn.net/zql1009/article/details/130735298
今日推荐