'MapTRv2' and 'CustomNuScenesOfflineLocalMapDataset' in MapTRv2

hustvl / MapTR

[ICLR'23 Spotlight] MapTR: Structured Modeling and Learning for Online Vectorized HD Map Construction

MIT License

984 stars 152 forks source link

'MapTRv2' and 'CustomNuScenesOfflineLocalMapDataset' in MapTRv2 #96

Closed sjtuxcx closed 10 months ago

sjtuxcx commented 10 months ago

Hi, thank you for your inspiring work! When I run the MapTRv2 code, I found the error of "KeyError: 'MapTRv2 is not in the models registry'" and "KeyError: 'CustomNuScenesOfflineLocalMapDataset is not in the dataset registry'". Are there some files missing?

LegendBC commented 10 months ago

You can use our latest code and checkout to maptrv2 branch. You can find the CustomNuScenesOfflineLocalMapDataset class here