The implementation of KDD2024 submitted paper "ControlTraj: Controllable Trajectory Generation with Topology-Constrained Diffusion Model"
The required packages with python environment is:
python>=3.7
torch>=1.7
pandas
numpy
matplotlib
pathlib
shutil
datetime
colored
math
-
Run and train the model by (The model will be saved for every 10 epoches):
python main.py