Skip to content

PyTorch implementation of paired img-to-img translation via SDE.

Notifications You must be signed in to change notification settings

Ishan-phys/paired-via-sde

Repository files navigation

Paired-Via-Sde

We implement a Score-based Generative Model to perform paired image-to-image translations from an image domain A to an image domain B and vice versa. Domain A consists of satellite images. Domain B consists of the corresponding maps style images.

Results:

1st column: Input Images / 2nd column: Ours (generated by conditional samplers) / 3rd column: Actual

We calculated the Structural Similarity Index Measure(SSIM) between the generated and actual images. With 1500 sampling steps the SSIM was calculated to be 0.977 showing that our model generated images very similar to the actual ground truth. The SSIM score as a function of the number of sampling steps is depicted in the figure below

SSIM vs Number of Sampling Steps

Dataset

The model was trained on the Maps dataset link.

Training

To set the parameters edit the config.py file and then run the code as given in the main.ipynb notebook file.

About

PyTorch implementation of paired img-to-img translation via SDE.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published