Traceback (most recent call last):
File “train.py”, line 51, in 
for i, data in enumerate(train_dataset):
File “/content/gdrive/My Drive/SNE-RoadSeg/data/init.py”, line 71, in iter
for i, data in enumerate(self.dataloader):
File “/usr/local/lib/python3.7/dist-packages/torch/utils/data/dataloader.py”, line 521, in next
data = self._next_data()
File “/usr/local/lib/python3.7/dist-packages/torch/utils/data/dataloader.py”, line 1203, in _next_data
return self._process_data(data)
File “/usr/local/lib/python3.7/dist-packages/torch/utils/data/dataloader.py”, line 1229, in _process_data
data.reraise()
File “/usr/local/lib/python3.7/dist-packages/torch/_utils.py”, line 425, in reraise
raise self.exc_type(msg)
cv2.error: Caught error 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 44, 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 44, in 
data = [self.dataset[idx] for idx in possibly_batched_index]
File “/content/gdrive/My Drive/SNE-RoadSeg/data/R2D_dataset.py”, line 38, in getitem
rgb_image = cv2.cvtColor(cv2.imread(os.path.join(useDir, ‘rgb’, name)), cv2.COLOR_BGR2RGB)
cv2.error: OpenCV(4.1.2) /io/opencv/modules/imgproc/src/color.cpp:182: error: (-215:Assertion failed) !_src.empty() in function ‘cvtColor’