Update README.md

README.md

@@ -14,6 +14,8 @@ tags:
 - 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
+- Example outputs
+  ![Example outputs](imgs/model_results.png)
 
 ## Model Details
 
@@ -40,6 +42,15 @@ The official PyTorch implementation [GitHub Repo](https://github.com/junyanz/pyt
 | 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) |
 
 
+**Loss Graphs**
+- Discriminator vs Epoch
+  ![Discriminator](imgs/lossD.jpeg)
+- Generator Overall Loss vs Epoch
+  ![Generator Overall Loss](imgs/lossG.jpeg)
+- Generator GAN Loss vs Epoch
+  ![Generator GAN Loss](imgs/lossG_GAN.jpeg)
+- Generator L1 Loss vs Epoch
+  ![Generator L1 Loss](imgs/lossG_L1.jpeg)
 <!-- 
 | 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)     |