open-mmlab / mmdetection3d

OpenMMLab's next-generation platform for general 3D object detection.
https://mmdetection3d.readthedocs.io/en/latest/
Apache License 2.0
5.2k stars 1.53k forks source link

[Docs] How to use custom dataset for training? #2603

Open sasuga62complex opened 1 year ago

sasuga62complex commented 1 year ago

Branch

dev-1.x branch https://mmdetection3d.readthedocs.io/en/dev-1.x/

šŸ“š The doc issue

I run the following command:

python tools/create_data.py custom --root-path ./data/custom --out-dir ./data/custom --extra-tag custom

in https://github.com/open-mmlab/mmdetection3d/blob/dev-1.x/docs/en/advanced_guides/customize_dataset.md. The command, however, went wrong with NotImplementedError: Don't support custom dataset. I think there is no related code about custom dataset format in 'tools/create_data.py`.

I want to use pure point cloud dataset and its annotated dataset prepared using SUSTechPoints. How can I do to use them to train pointpillars model?

Suggest a potential alternative/fix

No response

xiaodiadai commented 1 year ago

I also encountered this issue, there is no processing code for 'custom' in script tools/create_data.py

kekaodechenye commented 1 year ago

so do iļ¼ļ¼ļ¼

sunjiahao1999 commented 1 year ago

You need to emulate other datasets to generate train/test/val_infos.pkl, for pure point cloud datasets I recommend KITTI/ScanNet format for Outdoor/Indoor scenes.

sasuga62complex commented 1 year ago

@sunjiahao1999 Thanks. I already covert my point cloud datasets to KITTI data format.

dikubab commented 1 year ago

@sasuga62complex Can share the how you converted it?

xiaodiadai commented 1 year ago

@sasuga62complex Can share the how you converted it?

+1

calmelo commented 1 year ago

You need to emulate other datasets to generate train/test/val_infos.pkl, for pure point cloud datasets I recommend KITTI/ScanNet format for Outdoor/Indoor scenes.

How do you solve the problem? I can't use my customized datasets to train the mmdetection3d models. I would be very appreciate it if you can tell me how should I do it. Thanks!

HydrogenWasser commented 1 year ago

u need to rewrite data_converter.py and create_data.py for ur own datasets

linClubs commented 9 months ago

my annotated dataset prepared using SUSTechPoints ,and I already covert my point cloud datasets to KITTI data format. the next, due to my classes more than 3, how do rewrite data_converter.py and create_data? @sunjiahao1999 @HydrogenWasser

sasuga62complex commented 9 months ago

my annotated dataset prepared using SUSTechPoints ,and I already covert my point cloud datasets to KITTI data format. the next, due to my classes more than 3, how do rewrite data_converter.py and create_data? @sunjiahao1999 @HydrogenWasser

7 classes are already included for KITTI data format (I think it's related class KittiDataset in mmdet3d/datasets/kitti_datasets.py, maybe...) In addition, You need to check configs/base/datasets/*.py which you use.

18722013889 commented 7 months ago

You need to emulate other datasets to generate , for pure point cloud datasets I recommend KITTI/ScanNet format for Outdoor/Indoor scenes.train/test/val_infos.pkl when i convert mydataset to kitti, i have no images in mydataset , excute "python tools/create_data.py kitti --root-path .\data\mydata\ --out-dir .\data\mydata\ --extra-tag mydata" i got"file not exist: training\image_2\001.png" how can i solve thi problem.thank you!