mmdetection3d icon indicating copy to clipboard operation
mmdetection3d copied to clipboard

[Docs] How to use custom dataset for training?

Open sasuga62complex opened this issue 1 year ago • 12 comments

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

sasuga62complex avatar Jun 12 '23 14:06 sasuga62complex

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

xiaodiadai avatar Jun 15 '23 02:06 xiaodiadai

so do i!!!

kekaodechenye avatar Jun 25 '23 14:06 kekaodechenye

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.

sunjiahao1999 avatar Jun 28 '23 03:06 sunjiahao1999

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

sasuga62complex avatar Jun 28 '23 03:06 sasuga62complex

@sasuga62complex Can share the how you converted it?

dikubab avatar Jun 30 '23 07:06 dikubab

@sasuga62complex Can share the how you converted it?

+1

xiaodiadai avatar Jun 30 '23 07:06 xiaodiadai

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!

calmelo avatar Aug 03 '23 02:08 calmelo

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

HydrogenWasser avatar Sep 21 '23 09:09 HydrogenWasser

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

linClubs avatar Dec 08 '23 06:12 linClubs

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.

sasuga62complex avatar Dec 14 '23 04:12 sasuga62complex

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!

18722013889 avatar Feb 22 '24 02:02 18722013889