ocf-data-sampler

ocf-data-sampler contains all the tools needed to create batches and feed them to our models, such as PVNet. The data we work with—typically energy data, satellite imagery, and numerical weather predictions (NWPs)—is usually too heavy to do this on the fly, so that's where this repo comes in: handling steps like opening the data, selecting the right samples, normalising and reshaping, and saving to and reading from disk.

We are currently migrating to this repo from ocf_datapipes, which performs the same functions but is built around PyTorch DataPipes, which are quite cumbersome to work with and are no longer maintained by PyTorch. ocf-data-sampler uses PyTorch Datasets, and we've taken the opportunity to make the code much cleaner and more manageable.

Note

This repository is still in development and does not yet have the full functionality of its predecessor, ocf_datapipes. It might not be ready for use out of the box! We would really appreciate any help to let us make the transition faster.

Documentation

ocf-data-sampler doesn't have external documentation yet; you can read a bit about how our torch datasets work in the README here.

FAQ

If you have any questions about this or any other of our repos, don't hesitate to hop to our Discussions Page!

How does ocf-data-sampler deal with data sources that use different projections (e.g. some are in latitude-longitude, and some in OSGB)?

When creating samples, we make a spatial crop of a preset size centred around a point of interest (POI, usually a solar or wind farm). The size of the crop is set not in miles or kilometres, but in 'pixels', which would be different for different data sources, depending on their spatial resolution, projections they use, and where the POI is. For example, a latitude-longitude source with a 1° resolution will have pixel sizes corresponding to very different 'surface' distances (that you might measure in, e.g., kilometres) from a source with 0.1° resolution. The pixel size will even be different for the same source depending on how close the POI is to the equator!

Instead of trying to accommodate for all these differences and make all the sources use the same spatial grid, we translate the POI's position into the corresponding coordinate system and select the crop using the source's original grid. This 'snapshot' is then passed to the model with no additional information on what specific coordinates it represents; instead, since the size is always the same and the POI is always in the centre, the model gets consistent information on the measurements at a location near the POI and how it affects the target, without any explicit knowledge of where that location is in coordinate system terms.

Development

You can install ocf-data-sampler for development as follows:

pip install git+https://github.com/openclimatefix/ocf-data-sampler.git

Running the test suite

The tests in this project use pytest. Once you have it installed, you can run it from the project's directory:

cd ocf-data-sampler
pytest

Contributing and community

• PR's are welcome! See the Organisation Profile for details on contributing
• Find out about our other projects in the OCF Meta Repo
• Check out the OCF blog for updates
• Follow OCF on LinkedIn

Contributors

Thanks goes to these wonderful people (emoji key):

James Fulton 💻

Alexandra Udaltsova 💻

Sukhil Patel 💻

Peter Dudfield 💻

Vikram Pande 💻

This project follows the all-contributors specification. Contributions of any kind welcome!

---

Part of the Open Climate Fix community.