File size: 4,571 Bytes
8a63b96
 
 
 
 
 
 
 
 
 
 
 
e129637
 
 
 
feacf5b
8a63b96
 
 
 
 
 
 
 
 
 
 
 
8eb814b
 
8a63b96
ee78afb
 
 
 
 
 
 
 
2429981
7ac172d
 
795d905
 
 
 
7ac172d
8a63b96
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
---
license: mit
pipeline_tag: image-to-image
library_name: pytorch
tags:
- Pix2Pix
---

# 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](https://github.com/yuuIind/SAR2Optical)
- The demo: [yuulind/sar2rgb](https://huggingface.co/spaces/yuulind/sar2rgb)
- Includes checkpoints from epoch: 180, 265, 295

## Model Details

### Model Description

<!-- Provide a longer summary of what this model is. -->

This is custom implementation of pix2pix architecture in PyTorch. The code can found in [GitHub Repo](https://github.com/yuuIind/SAR2Optical)

The official lua implementaion: [GitHub Repo](https://github.com/phillipi/pix2pix)
The official PyTorch implementation [GitHub Repo](https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix)

- **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](https://huggingface.co/yuulind/pix2pix-sar2rgb/resolve/main/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](https://huggingface.co/yuulind/pix2pix-sar2rgb/resolve/main/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](https://huggingface.co/yuulind/pix2pix-sar2rgb/resolve/main/pix2pix_gen_295.pth)   |
| 4      | pix2pix_disc_180.pth    | Discriminator | Pix2Pix discriminator from epoch 180, with transpose convolution generator.       | [pix2pix_disc_180.pth](https://huggingface.co/yuulind/pix2pix-sar2rgb/resolve/main/pix2pix_disc_180.pth) |
| 5      | pix2pix_disc_265.pth    | Discriminator | Pix2Pix discriminator from epoch 265, with transpose convolution generator.       | [pix2pix_disc_265.pth](https://huggingface.co/yuulind/pix2pix-sar2rgb/resolve/main/pix2pix_disc_265.pth) |
| 6      | pix2pix_disc_295.pth    | Discriminator | Pix2Pix discriminator from epoch 295, with transpose convolution generator.       | [pix2pix_disc_295.pth](https://huggingface.co/yuulind/pix2pix-sar2rgb/resolve/main/pix2pix_disc_295.pth) |


<!-- 
| 4      | pix2pix_gen_nn_150.pth  | Generator     | Pix2Pix generator with upsampling, followed by normal convolutions, 150 epochs. | [pix2pix_gen_nn_150.pth](https://huggingface.co/yuulind/pix2pix-sar2rgb/resolve/main/pix2pix_gen_nn_150.pth)     |
| 5      | pix2pix_gen_nn_300.pth  | Generator     | Pix2Pix generator with upsampling, followed by normal convolutions, 300 epochs. | [pix2pix_gen_nn_300.pth](https://huggingface.co/yuulind/pix2pix-sar2rgb/resolve/main/pix2pix_gen_nn_300.pth)     |
| 8      | pix2pix_disc_nn_150.pth | Discriminator | Pix2Pix discriminator from epoch 150, with upsampling generator.                | [pix2pix_disc_nn_150.pth](https://huggingface.co/yuulind/pix2pix-sar2rgb/resolve/main/pix2pix_disc_nn_150.pth)   |
| 9      | pix2pix_disc_nn_300.pth | Discriminator | Pix2Pix discriminator from epoch 300, with upsampling generator.                | [pix2pix_disc_nn_300.pth](https://huggingface.co/yuulind/pix2pix-sar2rgb/resolve/main/pix2pix_disc_nn_300.pth)   |
-->
### Model Sources

<!-- Provide the basic links for the model. -->

- **Repository:** [GitHub Repo](https://github.com/yuuIind/SAR2Optical)
- **Paper:** [arxiv:1611.07004](https://arxiv.org/abs/1611.07004)
- **Official implementation:** [GitHub Repo](https://github.com/phillipi/pix2pix)

## Uses

<!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->

Converts SAR images to Optical (RGB) images