abduallahmohamed / Social-STGCNN

Code for "Social-STGCNN: A Social Spatio-Temporal Graph Convolutional Neural Network for Human Trajectory Prediction" CVPR 2020
MIT License
483 stars 141 forks source link

Details about Gaussian path visualization #72

Open ldx5972 opened 4 months ago

ldx5972 commented 4 months ago

Hello, Sir.Thank you for your work, and I am a student studying the field of trajectory prediction. I want to display the Gaussian path of my own test scenario based on your code, My idea is to obtain the pixel coordinates of pedestrians through the object detection algorithm, and then conduct data pre-processing, and then put the processed data into your Social-STGCNN network, and calculate the Gaussian distribution function of the returned results for visualization. The code for the Gauss part is as follows: 58KRPN{ Z~0{KA`MH54W236 I would like to ask you how you can display the predicted Gaussian path heat map in the test video, just like the effect in your paper. Thank you very much.