search

blakechen97 / SASA

SASA: Semantics-Augmented Set Abstraction for Point-based 3D Object Detection
Apache License 2.0
88 stars 14 forks source link

Custom dataset training ? #4

Open devendraswamy opened 2 years ago

devendraswamy commented 2 years ago

Please help me, is it possible to train the model with custom dataset , how can I train and inference the model with custom dataset , if possible please help me , Thank you so much in advance.

blakechen97 commented 2 years ago

Hi,

It is feasible to train or evaluate the model with custom datasets, which is supported by the OpenPCDet framework. You can check this document to learn how to transform your custom data to the format required by the framework.

OrangeSodahub commented 2 years ago

Hello, I've completed to train custom datasets (modify relative codes) and custom datasets has the label files similar to kitti. You can refer to https://github.com/OrangeSodahub/CRLFnet/blob/master/src/site_model/src/LidCamFusion/OpenPCDet/pcdet/datasets/custom/README.md Relative files are located in for example dataset/custom_dataset ... check by yourself!