ControlTraj

The implementation of KDD2024 research paper "ControlTraj: Controllable Trajectory Generation with Topology-Constrained Diffusion Model"

Requirements

The required packages with python environment is:

  python>=3.7
  torch>=1.7
  pandas
  numpy
  matplotlib
  pathlib
  shutil
  datetime
  colored
  math

Usage

1. Run and train the model by (The model will be saved for every 10 epoches):

   python main.py