SysCV / shift-detection-tta

This repository implements continuous test-time adaptation algorithms for object detection on the SHIFT dataset.
MIT License
18 stars 2 forks source link

KeyError: 'attributes' #9

Closed HEasoner closed 11 months ago

HEasoner commented 11 months ago

I finished get_start and dataset_prepare ,and run python tools/train.py configs/source/yolox/amp_yolox_x_8xb4-24e_shift_clear_daytime.py. Traceback (most recent call last): File "/root/miniconda3/envs/my-env/lib/python3.9/site-packages/mmengine/registry/build_functions.py", line 122, in build_from_cfg obj = obj_cls(args) # type: ignore File "/root/autodl-tmp/shift-detection-tta/shift_tta/datasets/shift_dataset.py", line 30, in init super().init(*args, *kwargs) File "/root/miniconda3/envs/my-env/lib/python3.9/site-packages/mmtrack/datasets/base_video_dataset.py", line 54, in init super().init(args, kwargs) File "/root/miniconda3/envs/my-env/lib/python3.9/site-packages/mmengine/dataset/base_dataset.py", line 245, in init self.full_init() File "/root/miniconda3/envs/my-env/lib/python3.9/site-packages/mmtrack/datasets/base_video_dataset.py", line 80, in full_init self.valid_data_indices = self.filter_data() File "/root/autodl-tmp/shift-detection-tta/shift_tta/datasets/shift_dataset.py", line 106, in filter_data self.valid_data_indices = self.filter_by_attributes() File "/root/autodl-tmp/shift-detection-tta/shift_tta/datasets/shift_dataset.py", line 41, in filter_by_attributes valid_data_indices = self._filter_image_by_attributes() File "/root/autodl-tmp/shift-detection-tta/shift_tta/datasets/shift_dataset.py", line 88, in _filter_image_by_attributes data_info['attributes'], self.filter_cfg['attributes'] KeyError: 'attributes'

During handling of the above exception, another exception occurred:

Traceback (most recent call last): File "/root/miniconda3/envs/my-env/lib/python3.9/site-packages/mmengine/registry/build_functions.py", line 122, in build_from_cfg obj = obj_cls(*args) # type: ignore File "/root/miniconda3/envs/my-env/lib/python3.9/site-packages/mmdet/datasets/dataset_wrappers.py", line 62, in init self.dataset = DATASETS.build(dataset) File "/root/miniconda3/envs/my-env/lib/python3.9/site-packages/mmengine/registry/registry.py", line 570, in build return self.build_func(cfg, args, **kwargs, registry=self) File "/root/miniconda3/envs/my-env/lib/python3.9/site-packages/mmengine/registry/build_functions.py", line 144, in build_from_cfg raise type(e)( KeyError: "class SHIFTDataset in shift_tta/datasets/shift_dataset.py: 'attributes'"

During handling of the above exception, another exception occurred:

Traceback (most recent call last): File "/root/miniconda3/envs/my-env/lib/python3.9/site-packages/mmengine/registry/build_functions.py", line 122, in build_from_cfg obj = obj_cls(*args) # type: ignore File "/root/miniconda3/envs/my-env/lib/python3.9/site-packages/mmengine/runner/loops.py", line 44, in init super().init(runner, dataloader) File "/root/miniconda3/envs/my-env/lib/python3.9/site-packages/mmengine/runner/base_loop.py", line 26, in init self.dataloader = runner.build_dataloader( File "/root/miniconda3/envs/my-env/lib/python3.9/site-packages/mmengine/runner/runner.py", line 1351, in build_dataloader dataset = DATASETS.build(dataset_cfg) File "/root/miniconda3/envs/my-env/lib/python3.9/site-packages/mmengine/registry/registry.py", line 570, in build return self.build_func(cfg, args, **kwargs, registry=self) File "/root/miniconda3/envs/my-env/lib/python3.9/site-packages/mmengine/registry/build_functions.py", line 144, in build_from_cfg raise type(e)( KeyError: 'class MultiImageMixDataset in mmdet/datasets/dataset_wrappers.py: "class SHIFTDataset in shift_tta/datasets/shift_dataset.py: \'attributes\'"'

During handling of the above exception, another exception occurred:

Traceback (most recent call last): File "/root/autodl-tmp/tools/train.py", line 126, in main() File "/root/autodl-tmp/tools/train.py", line 122, in main runner.train() File "/root/miniconda3/envs/my-env/lib/python3.9/site-packages/mmengine/runner/runner.py", line 1701, in train self._train_loop = self.build_train_loop( File "/root/miniconda3/envs/my-env/lib/python3.9/site-packages/mmengine/runner/runner.py", line 1493, in build_train_loop loop = LOOPS.build( File "/root/miniconda3/envs/my-env/lib/python3.9/site-packages/mmengine/registry/registry.py", line 570, in build return self.build_func(cfg, *args, **kwargs, registry=self) File "/root/miniconda3/envs/my-env/lib/python3.9/site-packages/mmengine/registry/build_functions.py", line 144, in build_from_cfg raise type(e)( KeyError: 'class EpochBasedTrainLoop in mmengine/runner/loops.py: \'class MultiImageMixDataset in mmdet/datasets/dataset_wrappers.py: "class SHIFTDataset in shift_tta/datasets/shift_dataset.py: \\'attributes\\'"\''