MTR icon indicating copy to clipboard operation
MTR copied to clipboard

not predicting trajectory for every agent; validation data

Open ahsan155 opened this issue 1 year ago • 3 comments

I trained the model and have checkpoints. I make prediction by running test.py in tools folder. I checked the config file and it looks like prediction is made on validation data. Running test.py creates output folder which contains result.pkl file. But the issue I have is, this result.pkl file does not contain predicted trajectory for all the agents that are in validation data for a particular scenario. In one scenario of validation data, I have 83 agents but for the same scenario, in result.pkl file there are only 3 agents with trajectory. How can I get trajectory for all agents?

ahsan155 avatar Oct 15 '23 05:10 ahsan155

Have you pulled the newest commit? Previous codes seem to have this issue, but the newest commit changes the data structure of result.pkl, which should have solve this.

FQYQC avatar Oct 19 '23 07:10 FQYQC

I also encountered this problem and it seems still not been solved.

BJHYZJ avatar Oct 24 '23 14:10 BJHYZJ

I figured it out. At the end of tfrecord file, there is field called tracks_to_predict which specifies the index of tracks to make prediction on. In my case, tracks_to_predict field had index of only 3 agents. I inserted more index in this field to do more predictions.

ahsan155 avatar Oct 24 '23 20:10 ahsan155