3D-CNN-Autoencoder

Powered by satolab

Overview

encoder-decoderモデルに3DCNNを組みこんだ，動画再構成モデルです． GRU-AEと比較した性能向上は見込めませんでした．

This is a video reconstruction model that combines 3D-CNN and encoder-decoder model.

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.)
main.pyで学習．logs/generated_videos/3dconvにサンプルが保存されます． Learn with main.py. The sample is saved in logs/generated_videos/3dconv.

動画ファイルのロード部分および3DCNNモデル定義部参考 https://github.com/DLHacks/mocogan

Name		Name	Last commit message	Last commit date
Latest commit History 10 Commits
logs		logs
README.md		README.md
lib.py		lib.py
main.py		main.py
network.py		network.py