mmpretrain使用报错：KeyError: 'img' 恳求解答

人工智能领域新星创作者

2023-06-07 22:25:46

原错误信息：

Original Traceback (most recent call last):
  File "/home/zhaocake/.conda/envs/mmlab/lib/python3.9/site-packages/torch/utils/data/_utils/worker.py", line 202, in _worker_loop
    data = fetcher.fetch(index)
  File "/home/zhaocake/.conda/envs/mmlab/lib/python3.9/site-packages/torch/utils/data/_utils/fetch.py", line 44, in fetch
    data = [self.dataset[idx] for idx in possibly_batched_index]
  File "/home/zhaocake/.conda/envs/mmlab/lib/python3.9/site-packages/torch/utils/data/_utils/fetch.py", line 44, in <listcomp>
    data = [self.dataset[idx] for idx in possibly_batched_index]
  File "/home/zhaocake/.conda/envs/mmlab/lib/python3.9/site-packages/mmengine/dataset/base_dataset.py", line 413, in __getitem__
    data = self.prepare_data(idx)
  File "/home/zhaocake/.conda/envs/mmlab/lib/python3.9/site-packages/mmengine/dataset/base_dataset.py", line 797, in prepare_data
    return self.pipeline(data_info)
  File "/home/zhaocake/.conda/envs/mmlab/lib/python3.9/site-packages/mmengine/dataset/base_dataset.py", line 59, in __call__
    data = t(data)
  File "/home/zhaocake/.conda/envs/mmlab/lib/python3.9/site-packages/mmcv/transforms/base.py", line 12, in __call__
    return self.transform(results)
  File "/home/zhaocake/PycharmProjects/mmpretrain/mmpretrain/datasets/transforms/processing.py", line 213, in transform
    img = results['img']
KeyError: 'img'

问题出现位置：

在加载完配置文件下载完与训练权重之后，出现上述报错。报错分为两部分，上述部分实际出现在下述部分之后：

Traceback (most recent call last):
  File "/home/zhaocake/PycharmProjects/mmpretrain/tools/train.py", line 159, in <module>
    main()
  File "/home/zhaocake/PycharmProjects/mmpretrain/tools/train.py", line 155, in main
    runner.train()
  File "/home/zhaocake/.conda/envs/mmlab/lib/python3.9/site-packages/mmengine/runner/runner.py", line 1721, in train
    model = self.train_loop.run()  # type: ignore
  File "/home/zhaocake/.conda/envs/mmlab/lib/python3.9/site-packages/mmengine/runner/loops.py", line 96, in run
    self.run_epoch()
  File "/home/zhaocake/.conda/envs/mmlab/lib/python3.9/site-packages/mmengine/runner/loops.py", line 111, in run_epoch
    for idx, data_batch in enumerate(self.dataloader):
  File "/home/zhaocake/.conda/envs/mmlab/lib/python3.9/site-packages/torch/utils/data/dataloader.py", line 517, in __next__
    data = self._next_data()
  File "/home/zhaocake/.conda/envs/mmlab/lib/python3.9/site-packages/torch/utils/data/dataloader.py", line 1199, in _next_data
    return self._process_data(data)
  File "/home/zhaocake/.conda/envs/mmlab/lib/python3.9/site-packages/torch/utils/data/dataloader.py", line 1225, in _process_data
    data.reraise()
  File "/home/zhaocake/.conda/envs/mmlab/lib/python3.9/site-packages/torch/_utils.py", line 429, in reraise
    raise self.exc_type(msg)
KeyError: Caught KeyError in DataLoader worker process 0.

副图片如下：