Check frame number and pedestrian id

[saruvora]

Hi, I want to check out the frame number and pedestrian id for the plots. How can I do that ?

[marsmarcin]

if you print the "pred_traj_fake " ,you can see the result of the Predicted trajectory while the input is "obs_traj", it is a batch of the dataset, not just one ID, if you really want to check it, you can just print one ,such as "pred_traj_gt[:,3,:].data","obs_traj[:,3,:].data",and find it in the text with "ctrl + F"

[saruvora]

Hi, thank you for your response. I tried this and got: tensor([[ 4.3400, 5.3000], [ 3.4900, 5.1300], [ 2.5800, 4.9600], [ 1.7400, 4.7600], [ 0.8900, 4.4000], [-0.0200, 4.1600], [-0.9700, 3.9500], [-1.6600, 3.8300], [-3.0400, 3.5400], [-3.9500, 3.3900], [-4.9200, 3.4500], [-5.9100, 3.5500]], device='cuda:0') input_a.data Out[3]: tensor([[12.2300, 6.6300], [11.2300, 6.6500], [10.0600, 6.7500], [ 9.1800, 6.5800], [ 8.1300, 6.4800], [ 7.2100, 6.0500], [ 6.2400, 5.7500], [ 5.2400, 5.5500]], device='cuda:0')

they are trajectory positions I want the pedestrian IDs and frame numbers