在colab里面跑pix2pixHD,增加了train_A,train_B,test_A和test_B四个文件夹,并且在train_A里面加了图片,出现这样的报错

(运行代码)
!python /content/drive/MyDrive/pix/pix2pixHD-master/train.py --name project1024 --label_nc 0 --no_instance --gpu_ids 0 --dataroot /content/drive/MyDrive/pix/pix2pixHD-master/datasets/cityscapes/
(错误提示)
Traceback (most recent call last):
  File "/content/drive/MyDrive/pix/pix2pixHD-master/train.py", line 60, in <module>
    for i, data in enumerate(dataset, start=epoch_iter):
  File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/dataloader.py", line 530, in __next__
    data = self._next_data()
  File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/dataloader.py", line 1224, in _next_data
    return self._process_data(data)
  File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/dataloader.py", line 1250, in _process_data
    data.reraise()
  File "/usr/local/lib/python3.7/dist-packages/torch/_utils.py", line 457, in reraise
    raise exception
IndexError: Caught IndexError in DataLoader worker process 0.
Original Traceback (most recent call last):
  File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/_utils/worker.py", line 287, in _worker_loop
    data = fetcher.fetch(index)
  File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/_utils/fetch.py", line 49, in fetch
    data = [self.dataset[idx] for idx in possibly_batched_index]
  File "/usr/local/lib/python3.7/dist-packages/torch/utils/data/_utils/fetch.py", line 49, in <listcomp>
    data = [self.dataset[idx] for idx in possibly_batched_index]
  File "/content/drive/MyDrive/pix/pix2pixHD-master/data/aligned_dataset.py", line 50, in __getitem__
    B_path = self.B_paths[index]
IndexError: list index out of range

能读取到数据么,检查一下代码,colab里面的目录引用和本地有点点不一样

你的图片最后一行“IndexError: list index out of range”说明了问题所在,列表索引超出范围了。在“B_path = self.B_paths[index]”这里设置断电,debug一下。出现这个问题的原因是变量B_path的长度小于变量self.B_paths[index]