Skip to content

nirraviv/RainGAN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

17 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

RainGAN

Applying simGAN for generating rainfall maps

Requirements

This repo tested on Python 3.6 and PyTorch 1.4.

The rest of the libraries are in requirements.txt

Dataset

The real dataset was downloaded from SMHI website (https://opendata.smhi.se/apidocs/radar/)

Real images dataset

  • Download to tiff images: python utils/dowload_radar_images.py (-h for parser arguments)
  • Preprocess: python utils/cut_radar_images.py (-h for parser arguments)

Synthetic images dataset

  • Run: python utils/random_gaussian_generation.py (-h for parser arguments)

GAN

All configuration parameters to train and evaluate RainGAN are in config.py

Train

  • In configuration file set is_train=True and is_eyes=False
  • Run: python main.py

Evaluate

  • In configuration file set load_from_iter, is_train=False and is_eyes=False
  • Run: python main.py

About

Applying simGAN for generating rainfall maps

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages