Forecasting with Swin-Unet on Satellite Meteorological Images

To use this forecasting model, install CUDA and PyTorch.

Additionally, install the required Python packages by running:

pip install -r requirements.txt

Download the datasets and organize them under the Dataset directory. Place each dataset under train, valid, and test directories accordingly.

To train the forecasting model, run the following command:

python training.py

To perform inference using the trained model, run the following command:

python inference.py