No description, website, or topics provided.
Python
Switch branches/tags
Nothing to show
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Failed to load latest commit information.
image
model
Import.py
MyLib.py
NetClass.py
README.md
graph_train_loss5001.png
train.py

README.md

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)
  2. Unsupervised Learning of Visual Structure using Predictive Generative Networks(http://arxiv.org/abs/1511.06380)