Pre-trained models

• ns-5m.pt: pre-trained model for the inverse problem of Navier-Stokes equation.
• blackhole-50k.pt: pre-trained model for the blackhole imaging problem.
• ffhq256.pt: pre-trained model for FFHQ256 image restoration tasks.

Training data for diffusion model

• navier-stokes-train.zip: data for training the diffusion model. All training samples have a cosine similarity of less than 0.63 compared to the test samples.

Data

• navier-stokes-test.zip: Navier-Stokes equation test data.
• bh-real.zip: blackhole test data.
• ffhq256-test.zip: FFHQ256 test data.