All Machine Learning

Pix2pix 高度图地形生成效果

使用的开源代码

https://github.com/affinelayer/pix2pix-tensorflow

 
 

安装

Conda create –name pix2pix

Conda activate pix2pix

Conda install cudatoolkit=9.0 tensorflow-gpu

 
 

训练命令:

Python pix2pix.py –mode train –outputdir heightmap_train –max_epochs 200 –input_dir heightmaps/train –which_direction AtoB

 
 

测试命令:

Python pix2pix.py –mode test –output_dir heightmaps_test –input_dir heightmaps/val –checkpoint heightmap_train

 
 

数据生成是由 GATA 那边搬过来的,直接可以用来执行。

 
 

效果展示【input/output/target