Yasoz / DiffTraj

The implemention of DiffTraj: Generating GPS Trajectory with Diffusion Probabilistic Model
MIT License
51 stars 10 forks source link

Is there any sample data? #18

Closed Qinzixin closed 1 month ago

Qinzixin commented 1 month ago

Hi,@Yasoz, l am very interested in your work.

unet = Guide UNet(config)#.cuda()
# print(unet)
traj = np.load('./xxxxxx',allow pickle=True)
traj = traj[:,:,:2]
head = np.load('./xxxxxx',allow pickle=True)

To better replicate the training process, could you pls provide any file of some sample data? Also, what's the difference between head and traj? It seems to me that head contains all tri points, and traj contains the the point's locations.

Thanks a lot!

Qinzixin commented 1 month ago

I just insert some random data, and the model works! Thanks a lot. The head actually means the trip attributes and the comment is very helpful!