Skip to content
This repository has been archived by the owner on Sep 28, 2023. It is now read-only.

PyTorch Dataloader for working with multi-modal data for nowcasting applications

License

Notifications You must be signed in to change notification settings

openclimatefix/nowcasting_dataloader

Repository files navigation

nowcasting-dataloader

codecov

PyTorch Dataloader for working with multi-modal data for nowcasting applications. In particular, this code loads the pre-prepared batches saved to disk by nowcasting_dataset. This code also computes some optional, additional input features.

Usage

Installation

Run pip install nowcasting-dataloader

Conventions

The dataloader assumes that data is generally in B, C, T, H, W ordering, where B is Batch size, C is number of channels, T is timestep, H is height, and W is width.