Skip to content

Kirili4ik/pix2pix-pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

14 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Pix2Pix model in PyTorch

This repository is an implementation of model described in Pix2Pix (Isola et al., 2016).

The task is to translate picture to a different modality (e.g. day photo to night one) using conditional GAN. Model was trained on facades and day2night datasets. Those can be found here.

There are some good examples of trained models:

alt text

alt text

Also there were some discoveries and experiements, e.g.:

alt text On the picture above dashed line represents loss on validation set. It's clearly seen that augmentations help to avoid overfitting.

More experiements can be found in my report (in Russian).