Pix2Pix Network

Learning process

Images from the test dataset at different epochs

Dataset

It uses its own dataset of satellite images from Bing and map schemes from OSM. See scrapper/data_scrapper.py

Prepare Weights

cd bin && ./merge_weight.sh
mv _weight.tar weight.tar

Usage

from net.pix2pix import Direction, Pix2Pix, Pix2PixTrainer

WEIGHT_PATH = "bin"
EPOCHS = 100

# Learn
p2p_tr = Pix2PixTrainer(".", "map_dataset", Direction.FORWARD, WEIGHT_PATH)
p2p_tr.train(EPOCHS, True)

# Or Test with weights loaded
p2p = Pix2Pix(".", "map_dataset", Direction.FORWARD, WEIGHT_PATH)
p2p.test(True)
# p2p.transfer_style(your_photo, True)