search

Vegeta2020 / SE-SSD

SE-SSD: Self-Ensembling Single-Stage Object Detector From Point Cloud, CVPR 2021.
Apache License 2.0
811 stars 128 forks source link

Custom Dataset #66

Closed gebawe closed 2 years ago

gebawe commented 2 years ago

Hi @Vegeta2020

Thank you for the great work.

Can you please include a custom dataset data loader with only point clouds and 3D bounding box (i.e Argoverse/ Waymo)?

Vegeta2020 commented 2 years ago

The code is mainly in preprocess.py and I simplify the code for other datasets, but you main find it in the repository Det3D that provides preprocess code for all datadets.