pix2pix-tensorflow实现了利用Tensorflow从一种图像转译成另一种图像的生成技术。基于pix2pix实现。pix2pix-tensorflow将pix2pix中的Torch移植为Tensorflow,同时包含来自Torch移植过来的色彩空间转换代码。
要求:Tensorflow0.12.1
推荐:LinuxwithTensorflowGPUedition+cuDNN
效果:
使用方法:
# clone this repogit clone https://github.com/affinelayer/pix2pix-tensorflow.gitcd pix2pix-tensorflow# download the CMP Facades dataset https://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 modelpython pix2pix.py --mode test --output_dir facades_test --input_dir facades/val --checkpoint facades_train# Resize source imagespython tools/process.py --input_dir photos/original --operation resize --output_dir photos/resized# Create images with blank centerspython tools/process.py --input_dir photos/resized --operation blank --output_dir photos/blank# Combine resized images with blanked imagespython tools/process.py --input_dir photos/resized --b_dir photos/blank --operation combine --output_dir photos/combined# Split into train/val setpython tools/split.py --dir photos/combined
评论