pix2pix-tensorflow 图像转译工具开源项目

我要开发同款
匿名用户2017年02月03日
33阅读
开发技术Python
所属分类程序开发、图形/图像处理
授权协议BSD

作品详情

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
声明:本文仅代表作者观点,不代表本站立场。如果侵犯到您的合法权益,请联系我们删除侵权资源!如果遇到资源链接失效,请您通过评论或工单的方式通知管理员。未经允许,不得转载,本站所有资源文章禁止商业使用运营!
下载安装【程序员客栈】APP
实时对接需求、及时收发消息、丰富的开放项目需求、随时随地查看项目状态

评论