Yasoz / DiffTraj

The implemention of DiffTraj: Generating GPS Trajectory with Diffusion Probabilistic Model

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Is there any sample data?

Qinzixin opened this issue · comments

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!

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!