search

m-hasan-n / pooling

"Pooling Toolbox" is the code of our work "Maneuver-Aware Pooling for Vehicle Trajectory Prediction".
MIT License
50 stars 8 forks source link

confused with the inputs #3

Closed chenqihuang closed 3 years ago

chenqihuang commented 3 years ago

First, this a wonderful research. I have a question about input of the network. In the code, the Input only includes the position, but not the radial velocity, is it right? Thank you for your valuable time.

m-hasan-n commented 3 years ago

Thanks for your interest in our work. The number of inputs depends on the selected pooling strategy. If the pooling strategy is set to 'polar', then 3 inputs are used (radius, angle and radial velocity). For all other strategies, only 2 inputs (x, y) are used. You can set the the pooling strategy args['pooling'] and number of inputs _args['ipdim'] in 'model_args.py' script.