waymo-open-dataset
waymo-open-dataset copied to clipboard
Where should I change the code if my gts' and pds' classes are diffrent with waymo proto?
Hi Waymo Research, thank you for providing this excellent dataset to the community.
I have used my own datasets, and there's more classes. But I'm used to this waymo-metrics way to evaluate my model locally. I have mapped several classes to VEHICLE, and other classes to CYCLITST. And it works well.
But I want to modify the code locally so that I can evaluate my own classes metrics. So any suggestions on how can I change the code?
Hi Sean,
I think you can do so by modifying the enum numbers in the label.proto files: https://github.com/waymo-research/waymo-open-dataset/blob/master/waymo_open_dataset/label.proto#L63-L69