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