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

Table 2: Speed (in seconds) comparison with S-LSTM #30

Open foxfromworld opened 5 years ago

foxfromworld commented 5 years ago

Hi

In your paper, the speed comparison is listed in Table 2. The speed of prediction length of 8 and 12 are 0.04 seconds and 0.05 seconds. May I ask how you calculate them? Because I tested them using code in evaluate_model.py listed below and found the averaged (20 times) time is between 0.007 and 0.009 seconds which are far lower than your result...

for _ in range(num_samples): pred_traj_fake_rel = generator( obs_traj, obs_traj_rel, seq_start_end ) pred_traj_fake = relative_to_abs( pred_traj_fake_rel, obs_traj[-1] ) ade.append(displacement_error( pred_traj_fake, pred_traj_gt, mode='raw' )) fde.append(final_displacement_error( pred_traj_fake[-1], pred_traj_gt[-1], mode='raw' )) print ("one time",toc - tic)

Thank you so much!