## Get Started

1. Install Python 3.8, PyTorch 1.9.0 for the code. 

2. Download data. [KTH action dataset](https://cloud.tsinghua.edu.cn/d/7d19372a621a4952b738/)

3. Train the model. You can use train.sh/train_base.sh and test.sh/test_base.sh to train and test the CPL-full/CPL-base model,   The learned model will be saved in the `--save_dir` folder. The generated future frames will be saved in the `--gen_frm_dir` folder.

**Please note: The KTH database is preprocessed and officially provided by an existing work (PredRNN-V2).**
