这个仓库旨在实现常用的视频预测算法,主要参考如下:
Prediction Under Uncertainty with Error-Encoding Networks EEN
- ConvLSTM
- PredNet prednet实现
- ...
- Moving MNIST
- UCF101
- ...
- pytorch
- ...
- Moving MNIST
- Kitti数据集kitti_dataset
- UCF101ucf101_dataset
- ...
可视化
# 在tmux或者另一个终端中开启可视化服务器visdom
python -m visdom.server
# 然后在浏览器中查看127.0.0.1:9097
训练
# 训练模型
python train.py
校验
# 校验模型
python validate.py
测试
# 测试模型
python test.py