waymo-open-dataset icon indicating copy to clipboard operation
waymo-open-dataset copied to clipboard

Where should I change the code if my gts' and pds' classes are diffrent with waymo proto?

Open sean-wade opened this issue 1 year ago • 1 comments

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?

sean-wade avatar Aug 22 '22 06:08 sean-wade

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

hfslyc avatar Aug 22 '22 17:08 hfslyc