wayveai / fiery

PyTorch code for the paper "FIERY: Future Instance Segmentation in Bird's-Eye view from Surround Monocular Cameras"
https://wayve.ai/blog/fiery-future-instance-prediction-birds-eye-view
MIT License
556 stars 85 forks source link

Visualization about distribution of futures #52

Closed Crayon-new closed 1 year ago

Crayon-new commented 1 year ago

Hi, thanks for your great work! I am very interested in the future distribution visualization in figure 1, could you please tell me how to plot it?

image
anthonyhu commented 1 year ago

Hello, to get this figure, I've generated different future trajectories by sampling a 100 times. Starting from a bird's-eye view image full of zeros, I then increment each cell whenever a trajectory passes through the cell. This image is then obtained by visualising the heatmap of the bird's-eye view cell counts.

If a trajectory is very likely, the cell count will be very high. Conversely, if a trajectory is very rare, it will have a low cell count.