search

wqi / WIMP

[arXiv] What-If Motion Prediction for Autonomous Driving ❓🚗💨
121 stars 24 forks source link

Can I get your inference/visualization code? #3

Open raspbe34 opened 3 years ago

raspbe34 commented 3 years ago

Hi, first of all, thanks for your awesome work and sharing that to us.

I tried to make inference/visualization code by myself, unfortunately, there were some problems.

Maybe library's mismatching, my insufficient coding skills, or something else.

So, can i get your inference/visualization code or even skeleton base code?

Yuanhf3754 commented 3 years ago

@raspbe34 Hi, I am a beginner, and I want to get inference/visualization code too. Did you solve this problem？Thank you! @wqi Could you help me?

raspbe34 commented 3 years ago

@raspbe34 Hi, I am a beginner, and I want to get inference/visualization code too. Did you solve this problem？Thank you!

@Yuanhf3754

I did it myself by using Argoverse Motion Forecasting Baselines and some Argoverse map api.

It's hard to give you the full code, but Argoverse Motion Forecasting Baselines will be helpful for you.

Also, because WIMP learns by pulling the training data back to ego vehicle, you have to put the WIMP's output back

to it's original coordinates by using reverse-translation and reverse-rotation if you want to use Argoverse map api.

There are absolute coordinate difference and rotation angle in input_dict.

Yuanhf3754 commented 3 years ago

@raspbe34
Maybe I just need a test code for submission and learning. Thank you for your reply, I will try it .