NVIDIA / earth2studio

Open-source deep-learning framework for exploring, building and deploying AI weather/climate workflows.
https://nvidia.github.io/earth2studio/
Apache License 2.0
73 stars 23 forks source link

Adding IMERG Data source #74

Closed NickGeneva closed 2 months ago

NickGeneva commented 2 months ago

Earth2Studio Pull Request

Description

Adding IMERG data source, the first data source using NASA's earth data portal. Validated output / transforms by comparing WB2 and IMERG at specific time. Both have the same larger structures indicating transforms are correct.

    from earth2studio.data import WB2ERA5

    date = datetime.datetime(year=2002, month=10, day=1)

    wbds = WB2ERA5(cache=False)
    imergds = IMERG(cache=False)

    wb_data = wbds(time=date, variable=["tp06"])
    im_data = imergds(time=date, variable=["tp"])

    import matplotlib.pyplot as plt
    from matplotlib.colors import LogNorm

    fig, ax = plt.subplots(1, 2)

    ax[0].imshow(wb_data.values[0, 0] + 0.001, norm=LogNorm(vmin=0.0001, vmax=1))
    ax[0].set_title("ERA5")
    ax[1].imshow(im_data.values[0, 0] + 0.001, norm=LogNorm(vmin=0.0001, vmax=1))
    ax[1].set_title("IMERG")

    plt.savefig("test.png")

test

Checklist

Dependencies

NickGeneva commented 2 months ago

/blossom-ci