agrimgupta92 / sgan

Code for "Social GAN: Socially Acceptable Trajectories with Generative Adversarial Networks", Gupta et al, CVPR 2018
MIT License
813 stars 261 forks source link

predict for single trajectory #40

Open saruvora opened 5 years ago

saruvora commented 5 years ago

Hi, I was just going through the trajectories.py while testing and generated trajectories for eth dataset along with frame and ped ids. I noticed that in trajectories.py if there are pedestrians present for example 2 and 3 the because of num_peds_considered it neglects pedestrian 2 and then uses pedestrian 3 to predict the pedestrian but then it does not detect anything for pedestrian 2. Can someone explain this? @agrimgupta92

PS- Can I generate predictions for each trajectory?