Diff-RNTraj

Source code of TKDE: Diff-RNTraj: A Structure-aware Diffusion Model for Road Network-constrained Trajectory Generation

Training:

Running the following command to train Diff-RNTraj.

Porto dataset: python multi_main.py --dataset Porto --diff_T 500 --pre_trained_dim 64 --rdcl 10

Chengdu dataset: python multi_main.py --dataset Chengdu --diff_T 500 --pre_trained_dim 64 --rdcl 10

Generate data:

After training the Diff-RNTraj, run the following command to generate the road network-constrained trajectory.

Porto dataset: python generate_data.py --dataset Porto --diff_T 500 --pre_trained_dim 64 --rdcl 10

Chengdu dataset: python generate_data.py --dataset Chengdu --diff_T 500 --pre_trained_dim 64 --rdcl 10

Generated Sample:

Our generated trajectories on the Chengdu and Porto datasets in the path of ./generate_data.

The processed dataset is available at https://drive.google.com/drive/folders/1TvotDt8wyZIJZg87_sBvYIi5wQVSO0OB?usp=drive_link