NO IMAGE
pix2pix-tensorflow 詳細介紹

pix2pix-tensorflow 實現了利用 Tensorflow 從一種影象轉譯成另一種影象的生成技術。基於 pix2pix 實現。pix2pix-tensorflow 將 pix2pix 中的 Torch 移植為 Tensorflow,同時包含來自 Torch 移植過來的色彩空間轉換程式碼。

源文連結:https://www.oschina.net/p/pix2pix-tensorflow

要求:Tensorflow 0.12.1

推薦:Linux with Tensorflow GPU edition cuDNN

效果:

使用方法:

# clone this repo
git clone https://github.com/affinelayer/pix2pix-tensorflow.git
cd pix2pix-tensorflow
# download the CMP Facades dataset http://cmp.felk.cvut.cz/~tylecr1/facade/
python tools/download-dataset.py facades
# train the model (this may take 1-8 hours depending on GPU, on CPU you will be waiting for a bit)
python pix2pix.py --mode train --output_dir facades_train --max_epochs 200 --input_dir facades/train --which_direction BtoA
# test the model
python pix2pix.py --mode test --output_dir facades_test --input_dir facades/val --checkpoint facades_train
# Resize source images
python tools/process.py --input_dir photos/original --operation resize --output_dir photos/resized
# Create images with blank centers
python tools/process.py --input_dir photos/resized --operation blank --output_dir photos/blank
# Combine resized images with blanked images
python tools/process.py --input_dir photos/resized --b_dir photos/blank --operation combine --output_dir photos/combined
# Split into train/val set

python tools/split.py –dir photos/combined