Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

https://github.com/kunimasa-kawasaki/chainer-prednet


https://github.com/kunimasa-kawasaki/chainer-prednet

Last synced: 9 days ago
JSON representation

Awesome Lists containing this project

README

        

# chainer-prednet
Deep Predictive Coding NetworksをChainerで実装を試みたもの。

# 中身
python train.py
train.py:学習用 NetClass.py:ネットワーク定義class
imageフォルダ:●が移動するサンプルデータです。
modelフォルダ:読み込み方法確認用です。(sampleの0~5000回の学習状態)

# 注意点
ConvLSTMを使用していないため参考文献2のCNN->LSTM->deCNNモデルです。
PredNetは階層性が特徴ですが、実装は1層のみとなっています。
制作途中のため正しく学習できるかどうかは不明です。
ubuntu14.04+spyderで開発してるためパス追加とか入ってます。

# 参考文献
1. Deep Predictive Coding Networks for Video Prediction and Unsupervised Learning(http://arxiv.org/abs/1605.08104)
3. Unsupervised Learning of Visual Structure using Predictive Generative Networks(http://arxiv.org/abs/1511.06380)