videopred

video prediction algorithms

这个仓库旨在实现常用的视频预测算法，主要参考如下：

• ConvLSTM_pytorch
• pytorch_convlstm
• ...

视频预测相关论文

Prediction Under Uncertainty with Error-Encoding Networks EEN

网络实现

• ConvLSTM
• PredNet prednet实现
• ...

数据集实现

• Moving MNIST
• UCF101
• ...

依赖

• pytorch
• ...

数据

• Moving MNIST
• Kitti数据集kitti_dataset
• UCF101ucf101_dataset
• ...

用法

可视化

visdom

# 在tmux或者另一个终端中开启可视化服务器visdom
python -m visdom.server
# 然后在浏览器中查看127.0.0.1:9097

训练

# 训练模型
python train.py

校验

# 校验模型
python validate.py

测试

# 测试模型
python test.py