-
Notifications
You must be signed in to change notification settings - Fork 42
Description
Traceback (most recent call last):
File "train.py", line 215, in
main()
File "train.py", line 112, in main
train_set, train_loader, train_sampler = build_dataloader(
File "/Private/Fusion/puzesong/DetZero/detection/detzero_det/datasets/init.py", line 44, in build_dataloader
dataset = all[dataset_cfg.DATASET](
File "/Private/Fusion/puzesong/DetZero/detection/detzero_det/datasets/waymo/waymo_dataset.py", line 21, in init
super().init(dataset_cfg, class_names, training, root_path, logger)
File "/Private/Fusion/puzesong/DetZero/detection/detzero_det/datasets/dataset.py", line 40, in init
self.data_augmentor = self.init_data_augmentor()
File "/Private/Fusion/puzesong/DetZero/detection/detzero_det/datasets/waymo/waymo_dataset.py", line 42, in init_data_augmentor
data_augmentor = DataAugmentor(
File "/Private/Fusion/puzesong/DetZero/detection/detzero_det/datasets/augmentor/data_augmentor.py", line 24, in init
cur_augmentor = getattr(self, cur_cfg.NAME)(config=cur_cfg)
File "/Private/Fusion/puzesong/DetZero/detection/detzero_det/datasets/augmentor/data_augmentor.py", line 28, in gt_sampling
db_sampler = database_sampler.DataBaseSampler(
File "/Private/Fusion/puzesong/DetZero/detection/detzero_det/datasets/augmentor/database_sampler.py", line 27, in init
[self.db_infos[cur_class].extend(infos[cur_class]) for cur_class in class_names]
File "/Private/Fusion/puzesong/DetZero/detection/detzero_det/datasets/augmentor/database_sampler.py", line 27, in
[self.db_infos[cur_class].extend(infos[cur_class]) for cur_class in class_names]
TypeError: list indices must be integers or slices, not str