agrimgupta92 / sgan

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

What does it mean by "seq_start_end" #18

Closed tracer9 closed 5 years ago

tracer9 commented 5 years ago

Hi, @agrimgupta92! Recently, I notice your social GAN work. When I read your code, I cannot understand "seq_start_end" in Decoder and PoolHiddenNet. I spent a lot of time and find out it has something about dataset. However, I still cannot understand the exact meaning of this parameter and why we should use it.

Could you give a gentle explanation? Thanks a lot.

jagjeet-singh commented 5 years ago

seq_start_end stores the start and end index of each sequence. If you have 3 sequences with 5 trajectories in each sequence, seq_start_end will store [(0,5),(5,10),(10,15)].