search

huang-xx / TrafficPredict

Pytorch implementation for "TrafficPredict: Trajectory Prediction for Heterogeneous Traffic-Agents"
109 stars 41 forks source link

Outputs look wrong #4

Open Binbose opened 5 years ago

Binbose commented 5 years ago

I trained the model, but when I am plotting the results, it looks like this trafpredict

This is not nearly as good as the images they show in the paper you reference though. Do you have a pretrained model with decent results?

Daniel-1597 commented 5 years ago

Till how many epochs you have trained, what is your best epoch what were the outputs you are plotting?

Binbose commented 5 years ago

I trained till epoch 300. The best epoch was 238. I tried both. I also tried plotting samples from the gaussian distribution, as well as only the predicted (mux,muy). I plotted by basically going in the get_mean_error_separately() method and scatter plot the ground truth and the prediction.

Daniel-1597 commented 5 years ago

we plotted next_x and next_y. our best epoch is 265? Do you know what is processed trajectory number?

Daniel-1597 commented 5 years ago

How can we visualize this on an image