實戰| 一行命令對你的圖像視頻進行風格遷移

1.項目介紹

今天我們要做的是一個快速圖像風格遷移的程序。

那麼，什麼是圖像風格遷移？圖像風格遷移就是把一種圖像風格轉變爲另一種圖像風格。例如，原圖爲：

加上不同風格的圖像可以得到如下不同的結果：

2.使用訓練好的模型來生成圖像

2.1環境

Python
Tensorflow

2.2模型下載

訓練好的模型有7個，表示7種類型的風格，模型文件的百度雲：
模型的百度雲地址
密碼:35pg

2.3使用訓練好的模型

在項目根目錄下執行：

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

--model_file 是模型的路徑，可以選擇7個模型中的一個
--image_file是原始圖片的路徑

新的圖片會存放在項目根目錄下：generated/res.jpg

3.訓練一個新的模型

3.1下載VGG16模型

如果要訓練一種新的圖像風格，可以先下載VGG16的模型：
VGG16模型
密碼:ykfy

然後在項目根目錄下新建一個名爲pretrained的文件夾，把vgg16的模型文件放入pretrained文件夾中。

3.2下載COCO數據集

下載地址
把解壓後的train2014文件夾放在項目根目錄下。

3.3創建新的yml文件

找一個新的風格的圖片，比如找一個火的圖片，路徑在img/fire.jpg。複製conf文件夾中wave.yml文件，然後改名fire.yml。把fire.yml中的：
style_image: img/wave.jpg
naming: "wave"
改爲
style_image: img/fire.jpg
naming: "fire"

3.4訓練新的圖像風格

python train.py -c conf/fire.yml

4.視頻的風格轉換

需要安裝opencv，安裝方式：
pip install opencv-python

準備好一個視頻文件，然後在項目根目錄下執行
python video.py --model_file models/wave.ckpt-done --video_file video/a.mp4

--model_file 是模型的路徑，可以選擇7個模型中的一個
--video_file是視頻文件的路徑

視頻效果

5.項目打包

大家可以去我的Github下載源代碼：
qbf的Github