A library for video frame prediction using PredRNN++, MIM, and Causal LSTM.
Project description
vPredicto
Predicto is a Python library for video frame prediction, featuring three state-of-the-art models: PredRNN++, MIM, and Causal LSTM. This library is designed to cater to both expert and non-expert users, providing an API for developers and a simple interface for non-experts.
Features
- Three video frame prediction models: PredRNN++, MIM, and Causal LSTM.
- Easy-to-use interface for training and testing models.
- Supports custom dataloaders or default to MovingMNIST dataset.
- Pre and post-processing for input and output in each model.
Installation
pip install vpredicto
Usage
Quick Start
from predicto import PredRNN, MIM, CausalLSTM, Predicto
# Create a model object
model_object = MIM()
# Initialize Predicto with the model object
model = Predicto(model_object)
# Train the model
model.train(train_loader)
# Test the model
model.test(test_loader)
Models
- PredRNN++: A recurrent neural network model for video frame prediction.
- MIM: Memory In Memory network for spatiotemporal predictive learning.
- Causal LSTM: A causal LSTM model for video frame prediction.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
vpredicto-0.1.34.tar.gz
(27.5 kB
view hashes)
Built Distribution
vpredicto-0.1.34-py3-none-any.whl
(46.6 kB
view hashes)
Close
Hashes for vpredicto-0.1.34-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 90a26d26b43b4c54d4c4103f11c7adc3e8acf73d77bbd921b2551fa9c2e5545e |
|
MD5 | fc2a6dc2e5ab39e809400306abe34c10 |
|
BLAKE2b-256 | 068eaf9bce6619df45c2037098a13a79077f0a266d86943c8414b6e0632f0ad3 |