When I was using part of your code about cityscapes benchmark, I met up with the error that
IndexError: Caught IndexError in DataLoader worker process 0.
Original Traceback (most recent call last):
File "/usr/local/lib/python3.6/dist-packages/torch/utils/data/_utils/worker.py", line 185, in _worker_loop
data = fetcher.fetch(index)
File "/usr/local/lib/python3.6/dist-packages/torch/utils/data/_utils/fetch.py", line 44, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/usr/local/lib/python3.6/dist-packages/torch/utils/data/_utils/fetch.py", line 44, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "../utils/datasets.py", line 155, in getitem
img1, target1 = self.transforms(img, target)
File "../utils/transforms.py", line 27, in call
image, target = t(image, target)
File "../utils/transforms.py", line 216, in call
target = target if type(target) == str else self.label_id_map[target]
IndexError: index 255 is out of bounds for dimension 0 with size 34
It seems that the LabelMap(label_id_map_city), didn't work correctly.
It's the first time to using this benchmark, so I dont know how to deal with this problem, could you plz give me some hints?