Skip to content
No description, website, or topics provided.
Branch: master
Clone or download
Latest commit 240cf6e Jul 20, 2018
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
data updated data loading to allow analogy plot to work. Feb 22, 2018
datasets kth download scripts. Feb 15, 2018
models proper lstm code. Feb 16, 2018
pretrained_models
.gitignore pretrained kth model. Jul 20, 2018
README.md Merge branch 'master' of https://github.com/edenton/drnet-py Feb 15, 2018
train_drnet.py pretrained kth model. Jul 20, 2018
train_lstm.py pretrained kth model. Jul 20, 2018
utils.py faster data loading for drnet training. Feb 22, 2018

README.md

drnet-py

Training on KTH

First download the KTH action recognition dataset by running:

sh datasets/download_kth.sh /my/kth/data/path/

where /my/kth/data/path/ is the directory the data will be downloaded into. Next, convert the downloaded .avi files into .png's for the data loader. To do this you'll want ffmpeg installed. The following script will do the conversion, but beware, it's written in lua:

th datasets/convert_kth.lua --dataRoot /my/kth/data/path/ --imageSize 128

The --imageSize flag specifiec the image resolution. Experimental results in the paper used 128x128, but you can also train a model on 64x64 and it will train much faster.

fyi

I will be most responsive regarding questions about the code if you email me at denton@cs.nyu.edu.

You can’t perform that action at this time.