pix2pix-tensorflow图像转译工具

联合创作 · 2023-09-29 16:33

pix2pix-tensorflow 实现了利用 Tensorflow 从一种图像转译成另一种图像的生成技术。基于 pix2pix 实现。pix2pix-tensorflow 将 pix2pix 中的 Torch 移植为 Tensorflow,同时包含来自 Torch 移植过来的色彩空间转换代码。

要求: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
浏览 4
点赞
评论
收藏
分享

手机扫一扫分享

编辑 分享
举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

编辑 分享
举报