玩一玩Neural Style

2017-09-02

尝试了一下机器学习的乐趣

一、TensorFlow Windows 下安装

  • 我选择的是 Anaconda 的安装方法,他会自动安装 Python 环境,我用的是 64 位的 Anaconda3 4.1.0,Python3.5.1
  • 之后安装 TensorFlow
  • Pip install tensorflow

  • or
  • Pip install tensorflow-gpu

  • pip 可能需要更新
  • 下的可能比较慢,也可以去 github 上 clone 一个。
  • 因为我有 N 卡,所以 CPU 版本和 GPU 版本都试了一下。GPU 版本需要安装 CUDA8 和 cuDnn 库,库文件放在 CUDA8 的安装目录下,而且需要继续添加环境变量。

二、neural-style

  • neural-style 是一个风格学习项目,好像是后来搬到 TensorFlow 上的。
  • 项目链接
  • neural-style 依赖 VGG19 的深度学习库(Pre-trained VGG network 在项目主页有提供下载)VGG19 主页
  • 之前 GPU 版本的环境变量一直配置不好,所以最后用了 CPU 版本。

  • python neural_style.py –content /content file/ –styles /style file/ –output /output file/

  • 在项目下填入训练风格输出文件等就 OK 了

image

  • 我用蒙娜丽莎做风格样本学习,在对梦露做风格处理,看上去效果一般,也许调教一下小参数会更好,但是笔记本的低压 U 在 iterations 设为 1000 时跑了 6 个小时 emmm…