search

gaozhihan / PreDiff

[NeurIPS 2023] Official implementation of "PreDiff: Precipitation Nowcasting with Latent Diffusion Models"
Apache License 2.0
102 stars 4 forks source link

How to obtain SSIM, CRPS, and FVD? #7

Open precipitation-zy opened 11 months ago

precipitation-zy commented 11 months ago

How to obtain SSIM, CRPS, and FVD? Will their codes be released later? I am looking forward to your response.

gaozhihan commented 11 months ago

Thank you for your interest and your question! We plan to integrate SSIM and FVD into the PreDiff training script via torchmetrics.Metric. For CRPS, we referred to the Pyro implementation to get the results reported in our paper. We are currently exploring options for integrating CRPS in a clean, seamless way.

gaozhihan commented 11 months ago

The integration of SSIM and FVD is done in #10