Details about Gaussian path visualization

[ldx5972]

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:

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.