search

HansBambel / SmaAt-UNet

PyTorch-Code for the Paper "SmaAt-UNet: Precipitation Nowcasting using a Small, Attentive UNet-Architecture"
230 stars 43 forks source link

Add mse per pixel #29

Closed HansBambel closed 6 months ago

HansBambel commented 6 months ago

This PR makes calc_metrics_test_set.py the main script for calculating the metrics. It is just missing the persistence model.

The metrics now get saved in a json file instead of a pickle file.