search

yuantianyuan01 / StreamMapNet

GNU General Public License v3.0
202 stars 17 forks source link

About training the model on different datasets #36

Open BaophanN opened 2 months ago

BaophanN commented 2 months ago

Hi author, there is a new dataset from OpenDriveLab which is called OpenLaneV2 which was built from NuScenes and AgoverseV2. They add some annotations for the task of topology reasoning. How can I adapt StreamMapNet to train on this dataset? Can you give me an overview on where I can get started? Thank you.

yuantianyuan01 commented 2 months ago

Hi, thank you for your interest in our work.

To adapt our method on the new dataset, you can first add a new target category lane centerline to detect the lanes. As for the topology, we model each map element as an individual query so you can add a new head to predict the connectivity matrix on these queries.