wayveai / mile

PyTorch code for the paper "Model-Based Imitation Learning for Urban Driving".
MIT License
355 stars 34 forks source link

why is the current step, not the next step? #30

Open wzn0828 opened 1 year ago

wzn0828 commented 1 year ago

Why predict the current step's observation and BEV segmentation? Why not directly predict the observation and segmentation of the next step? If so, imagination was not needed, one can directly use the prediction of image as the input for the next step. The model architecture would be simpler.