eclipse-pytorch 0.0.8

A pytorch implementation of Eclipse

Eclipse

Implementing Paletta et al in Pytorch

Most of the codebase comes from Fiery

Install

pip install eclipse_pytorch

How to use

import torch

from eclipse_pytorch.model import Eclipse

eclipse = Eclipse(horizon=5)

let's simulte some input images:

images = [torch.rand(2, 3, 128, 128) for _ in range(4)]

preds = eclipse(images)

you get a dict with forecasted masks and irradiances:

len(preds['masks']), preds['masks'][0].shape, preds['irradiances'].shape

(6, torch.Size([2, 4, 128, 128]), torch.Size([2, 6]))

Citation

@article{paletta2021eclipse,
  title     = {{ECLIPSE} : Envisioning Cloud Induced Perturbations in Solar Energy},
  author    = {Quentin Paletta and Anthony Hu and Guillaume Arbod and Joan Lasenby},
  year      = {2021},
  eprinttype = {arXiv},
  eprint    = {2104.12419}
}

Contribute

This repo is made with nbdev, please read the documentation to contribute