PJLab-ADG / 3DTrans

An open-source codebase for exploring autonomous driving pre-training
https://bobrown.github.io/Team_3DTrans.github.io/
Apache License 2.0
585 stars 72 forks source link

About running on our own Dataset #9

Closed deqwin closed 1 year ago

deqwin commented 1 year ago

Q1 How to perform the model training and inference using our own Dataset (for DA Waymo->our Dataset)?

Q2 How to write a new dataloader to load our private Dataset?

BOBrown commented 1 year ago

Thanks for your question

A1: you can write the dataloader function in pcdet/datasets, and retrain the model on Waymo dataset and perform the inference process.

A2: Due to that our 3DTrans is developed based on OpenPCDet, you can organize your dataset file and add the dataloader according to the guide in CUSTOM_DATASET