A pytorch implementation of Eclipse
Project description
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
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
eclipse_pytorch-0.0.5.tar.gz
(15.7 kB
view hashes)
Built Distribution
Close
Hashes for eclipse_pytorch-0.0.5-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 71766a05e20e9c738ff50660339a770b302cfc7e3acf6d9e664b4b05c286362b |
|
MD5 | 1d2425ba02080746b88f819023f67b3c |
|
BLAKE2b-256 | 521432af42e89ea87b2b9ccc9e361dc877fff32daf8d391e3b5bba2ed2dd5d72 |