junjie18 / CMT

[ICCV 2023] Cross Modal Transformer: Towards Fast and Robust 3D Object Detection
Other
308 stars 34 forks source link

CustomNuscenesDataset发生异常 #87

Closed BadNaive closed 8 months ago

BadNaive commented 8 months ago

junjie你好!我想咨询大佬一个问题,还望大佬能够抽空解答~

我将我的模型与我自己的数据按照nuscenes的格式进行了适配,但是在训练的时候报错如下: image

我的理解是可能我的自己的数据里面缺少construction_vehicle这一个类别导致CBGS报错,请问是这个原因吗?万分感谢!

以下是报错内容文字版(以防图片无法加载):

···

'CustomNuScenesDataset: \'UnifiedObjectSample: "UnifiedDataBaseSampler: \\'construction_vehicle\\'"\'' File "/workspace/Code/bevod_cmt/projects/mmdet3d_plugin/datasets/pipelines/dbsampler.py", line 59, in init db_infos = getattr(self, prep_func)(db_infos, val) File "/workspace/Code/bevod_cmt/projects/mmdet3d_plugin/datasets/pipelines/dbsampler.py", line 119, in filter_by_min_points for info in db_infos[name]:

During handling of the above exception, another exception occurred:

File "/workspace/Code/bevod_cmt/projects/mmdet3d_plugin/datasets/pipelines/transform_3d.py", line 128, in init self.db_sampler = build_from_cfg(db_sampler, OBJECTSAMPLERS)

During handling of the above exception, another exception occurred:

File "/workspace/Code/bevod_cmt/projects/mmdet3d_plugin/datasets/custom_nuscenes_dataset.py", line 24, in init super(CustomNuScenesDataset, self).init(*args, **kwargs) File "/workspace/Code/bevod_cmt/mmdetection3d/mmdet3d/datasets/nuscenes_dataset.py", line 131, in init super().init( File "/workspace/Code/bevod_cmt/mmdetection3d/mmdet3d/datasets/custom_3d.py", line 99, in init self.pipeline = Compose(pipeline) File "/workspace/Code/bevod_cmt/mmdetection3d/mmdet3d/datasets/pipelines/compose.py", line 31, in init transform = build_from_cfg(transform, MMDET_PIPELINES)

During handling of the above exception, another exception occurred:

File "/workspace/Code/bevod_cmt/mmdetection3d/mmdet3d/datasets/builder.py", line 46, in build_dataset dataset = build_from_cfg(cfg, MMDET_DATASETS, default_args) File "/workspace/Code/bevod_cmt/mmdetection3d/mmdet3d/datasets/builder.py", line 40, in build_dataset dataset = CBGSDataset(build_dataset(cfg['dataset'], default_args)) File "/workspace/Code/bevod_cmt/tools/train.py", line 257, in main datasets = [build_dataset(cfg.data.train)] File "/workspace/Code/bevod_cmt/tools/train.py", line 297, in main()

···

BadNaive commented 8 months ago

测试了一下,确实如此