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['irradiance'][0].shape
(6, torch.Size([2, 3, 128, 128]), torch.Size([2, 1]))
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.3.tar.gz
(14.8 kB
view hashes)
Built Distribution
Close
Hashes for eclipse_pytorch-0.0.3-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | c00620ae38d2b60e65553919f456fad6d2578089722802f7b561be40e4d36a51 |
|
MD5 | 5b9cc8384a6805f46b2cbe46bd24c15b |
|
BLAKE2b-256 | a146747d0c5c5bc330d31f7e07432e55e74b8f0d2a11740a31d4c8f885aa78bf |