RainUNet for Super-Resolution Rain Movie Prediction under Spatio-temporal Shifts

December 07, 2022 · Entered Twilight · 🏛 arXiv.org

Repo contents: .gitignore, COPYING, LICENSE, README.md, environment.yml, mk_env.sh, mk_heldout_core.sh, mk_heldout_transfer.sh, mk_pred_core.sh, mk_pred_core_3boxi2019.sh, mk_pred_transfer.sh, models, train.py, utils

Authors Jinyoung Park, Minseok Son, Seungju Cho, Inyoung Lee, Changick Kim arXiv ID 2212.04005 Category cs.CV: Computer Vision Citations 4 Venue arXiv.org Repository https://github.com/jinyxp/Weather4cast-2022 ⭐ 10 Last Checked 3 months ago

Abstract

This paper presents a solution to the Weather4cast 2022 Challenge Stage 2. The goal of the challenge is to forecast future high-resolution rainfall events obtained from ground radar using low-resolution multiband satellite images. We suggest a solution that performs data preprocessing appropriate to the challenge and then predicts rainfall movies using a novel RainUNet. RainUNet is a hierarchical U-shaped network with temporal-wise separable block (TS block) using a decoupled large kernel 3D convolution to improve the prediction performance. Various evaluation metrics show that our solution is effective compared to the baseline method. The source codes are available at https://github.com/jinyxp/Weather4cast-2022