Skip to content

satolab12/GRU-Autoencoder

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

24 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

GRU-Autoencoder

https://qiita.com/satolab/items/c4e9a287e8d1ebb010a6

Powered by satolab

Overview

時系列モデルであるGRUと,encoder-decoderモデルを組み合わせた,動画再構成モデルです. ここではこのモデルを,GRU-AEと呼びます.

This is a video reconstruction model that combines GRU (a time-series model) and encoder-decoder model. Here, we call this model GRU-AE.

Model

Results

  • 5,000 itr(input,output)

real_itr5000_no0.png recon_itr5000_no0.png

Usage

  • datasetのダウンロード Download dataset http://www.wisdom.weizmann.ac.il/~vision/SpaceTimeActions.html

  • lib.pyのParseGRU()内の初期化メソッド,dataset変数に, 上記のdatasetが格納されたdirを指定してください (動画ファイルのままで問題ございません) Please specify the dir in which the above dataset is stored. (No problem as a video file.)

  • gru_ae.pyで学習.logs/generated_videosにサンプルが保存されます. Learn with gru_ae.py. The sample is saved in logs/generated_videos.

参考文献 References

動画ファイルのロード部分 https://github.com/DLHacks/mocogan