mmyolo报错:ValueError: Key img_path is not in available keys
错误信息如下:
/home/wwy/miniconda3/envs/fomo/lib/python3.8/site-packages/albumentations/__init__.py:13: UserWarning: A new version of Albumentations is available: 2.0.8 (you have 1.4.18). Upgrade using: pip install -U albumentations. To disable automatic update checks, set the environment variable NO_ALBUMENTATIONS_UPDATE to 1.check_for_updates()
/home/wwy/miniconda3/envs/fomo/lib/python3.8/site-packages/albumentations/core/composition.py:192: UserWarning: Got processor for bboxes, but no transform to process it.self._set_keys()
/home/wwy/miniconda3/envs/fomo/lib/python3.8/site-packages/albumentations/core/composition.py:192: UserWarning: Got processor for bboxes, but no transform to process it.self._set_keys()
Traceback (most recent call last):File "third_party/mmyolo/tools/train.py", line 123, in <module>main()File "third_party/mmyolo/tools/train.py", line 119, in mainrunner.train()File "/home/wwy/miniconda3/envs/fomo/lib/python3.8/site-packages/mmengine/runner/runner.py", line 1777, in trainmodel = self.train_loop.run() # type: ignoreFile "/home/wwy/miniconda3/envs/fomo/lib/python3.8/site-packages/mmengine/runner/loops.py", line 98, in runself.run_epoch()File "/home/wwy/miniconda3/envs/fomo/lib/python3.8/site-packages/mmengine/runner/loops.py", line 114, in run_epochfor idx, data_batch in enumerate(self.dataloader):File "/home/wwy/miniconda3/envs/fomo/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 633, in __next__data = self._next_data()File "/home/wwy/miniconda3/envs/fomo/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 1345, in _next_datareturn self._process_data(data)File "/home/wwy/miniconda3/envs/fomo/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 1371, in _process_datadata.reraise()File "/home/wwy/miniconda3/envs/fomo/lib/python3.8/site-packages/torch/_utils.py", line 644, in reraiseraise exceptionValueError: Caught ValueError in DataLoader worker process 0.Original Traceback (most recent call last):File "/home/wwy/miniconda3/envs/fomo/lib/python3.8/site-packages/torch/utils/data/_utils/worker.py", line 308, in _worker_loopdata = fetcher.fetch(index)File "/home/wwy/miniconda3/envs/fomo/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 51, in fetchdata = [self.dataset[idx] for idx in possibly_batched_index]File "/home/wwy/miniconda3/envs/fomo/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 51, in <listcomp>data = [self.dataset[idx] for idx in possibly_batched_index]File "/home/wwy/miniconda3/envs/fomo/lib/python3.8/site-packages/mmengine/dataset/base_dataset.py", line 410, in __getitem__data = self.prepare_data(idx)File "/home/wwy/miniconda3/envs/fomo/lib/python3.8/site-packages/mmyolo/datasets/yolov5_coco.py", line 53, in prepare_datareturn self.pipeline(data_info)File "/home/wwy/miniconda3/envs/fomo/lib/python3.8/site-packages/mmcv/transforms/base.py", line 12, in __call__return self.transform(results)File "/home/wwy/miniconda3/envs/fomo/lib/python3.8/site-packages/mmcv/transforms/wrappers.py", line 88, in transformresults = t(results) # type: ignoreFile "/home/wwy/miniconda3/envs/fomo/lib/python3.8/site-packages/mmcv/transforms/base.py", line 12, in __call__return self.transform(results)File "/home/wwy/miniconda3/envs/fomo/lib/python3.8/site-packages/mmdet/structures/bbox/box_type.py", line 267, in wrapperreturn func(self, results)File "/home/wwy/miniconda3/envs/fomo/lib/python3.8/site-packages/mmdet/datasets/transforms/transforms.py", line 1699, in transformresults = self.aug(**results)File "/home/wwy/miniconda3/envs/fomo/lib/python3.8/site-packages/albumentations/core/composition.py", line 346, in __call__self.preprocess(data)File "/home/wwy/miniconda3/envs/fomo/lib/python3.8/site-packages/albumentations/core/composition.py", line 373, in preprocessraise ValueError(msg)ValueError: Key img_path is not in available keys.
解决方法:
降级 albumentations 版本。这个错误大概率是 albumentations 库的版本过高导致。可以通过降级到 1.3.1 版本来解决。
pip install albumentations==1.3.1
