Pix2Pix SAR To RGB

yuulind/pix2pix-sar2rgb

  • This is a Pix2Pix CGAN implementation for translating Synthetic Aperture Radar (SAR) images to Optical (RGB) images.
  • For more information on code, please check GitHub Repo
  • The demo: yuulind/sar2rgb
  • Includes checkpoints from epoch: 180, 265, 295
  • Example outputs Example outputs

Model Details

Model Description

This is custom implementation of pix2pix architecture in PyTorch. The code can found in GitHub Repo

The official lua implementaion: GitHub Repo The official PyTorch implementation GitHub Repo

  • Model type: Image to Image Translation
  • BASELINE FID SCORE is between 30 and 40
  • Baseline score is calculated by randomly sampling two sets from real images and comparing the them.
Number Name Model Type Description Link to Model File
1 pix2pix_gen_180.pth Generator Pix2Pix generator with transpose convolution, 180 epochs. FID score is 185.04 pix2pix_gen_180.pth
2 pix2pix_gen_265.pth Generator Pix2Pix generator with transpose convolution, 265 epochs. FID score is 189.81 pix2pix_gen_265.pth
3 pix2pix_gen_295.pth Generator Pix2Pix generator with transpose convolution, 295 epochs. FID score is 187.73 pix2pix_gen_295.pth
4 pix2pix_disc_180.pth Discriminator Pix2Pix discriminator from epoch 180, with transpose convolution generator. pix2pix_disc_180.pth
5 pix2pix_disc_265.pth Discriminator Pix2Pix discriminator from epoch 265, with transpose convolution generator. pix2pix_disc_265.pth
6 pix2pix_disc_295.pth Discriminator Pix2Pix discriminator from epoch 295, with transpose convolution generator. pix2pix_disc_295.pth

Loss Graphs

  • Discriminator vs Epoch Discriminator
  • Generator Overall Loss vs Epoch Generator Overall Loss
  • Generator GAN Loss vs Epoch Generator GAN Loss
  • Generator L1 Loss vs Epoch Generator L1 Loss

Model Sources

Uses

Converts SAR images to Optical (RGB) images

Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model is not currently available via any of the supported Inference Providers.
The model cannot be deployed to the HF Inference API: The HF Inference API does not support image-to-image models for pytorch library.