woodfrog / maptracker

Code for paper "MapTracker: Tracking with Strided Memory Fusion for Consistent Vector HD Mapping", ECCV 2024 (Oral)
https://map-tracker.github.io/
Other
144 stars 10 forks source link

Adding centerline category #12

Closed 304886938 closed 2 months ago

304886938 commented 2 months ago

Your work is wonderful.

I want to add centerline category like to MapTRv2. It seems that you provide the centerline annotation generation code in /maptracker-main/plugin/datasets/map_utils/nuscmap_extractor.py. Can I simply add the category to the config file ? V3UXIR4FSQDQ T{PP65@F T

woodfrog commented 2 months ago

Thanks for the question. That part of the code was inherited from MapTRv2's preprocessing, but we did not modify it for use in our method, as we followed the old benchmark that includes only the three map classes. If you want to include centerlines in your experiments, you will need to adjust the code and rerun the training.