InTraGen: Trajectory-controlled Video Generation for Object Interactions
_{PyTorch Implementation}

Paper

How to Run

Dowload dataset

MoVi-Extended | Domino | Football | Pool

1. Put and etract the dataset into datasets folder

2. Install the Environment

bash install.sh

3. Download the Pre-trained Checkpoints from Open-Sora-Plan, and save the Open-Sora-Plan-v1.1.0 into models folder

4. Run the Training

bash train.sh

Acknowledgement

Open-Sora-Plan