wayveai / mile

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

birdview pictures are all black, is it normal? #24

Open beyondpzk opened 11 months ago

beyondpzk commented 11 months ago

image

anthonyhu commented 11 months ago

Yes because they are saved as unique ID from [0, n]. To visualise them with colours you need to convert those ID to RGB values using https://github.com/wayveai/mile/blob/main/mile/constants.py#L24