Ski2DPose dataset

[umaiskhan42]

Hi, Loving your work!
Im using Ski2DPose dataset that i have already made in coco dataset format. However i have run into issues please check:

in train(model, dataloader, optimizer, epoch, iteration)
49 averMeters.clear()
50 end = time.time()
---> 51 for i, inputs in enumerate(dataloader):
52 averMeters['data_time'].update(time.time() - end)
53 iteration += 1

/usr/local/lib/python3.10/dist-packages/torch/utils/data/dataloader.py in next(self)
628 # TODO(pytorch/pytorch#76750)
629 self._reset() # type: ignore[call-arg]
--> 630 data = self._next_data()
631 self._num_yielded += 1
632 if self._dataset_kind == _DatasetKind.Iterable and \

/usr/local/lib/python3.10/dist-packages/torch/utils/data/dataloader.py in _next_data(self)
1343 else:
1344 del self._task_info[idx]
-> 1345 return self._process_data(data)
1346
1347 def _try_put_index(self):

/usr/local/lib/python3.10/dist-packages/torch/utils/data/dataloader.py in _process_data(self, data)
1369 self._try_put_index()
1370 if isinstance(data, ExceptionWrapper):
-> 1371 data.reraise()
1372 return data
1373

/usr/local/lib/python3.10/dist-packages/torch/_utils.py in reraise(self)
692 # instantiate since we don't know how to
693 raise RuntimeError(msg) from None
--> 694 raise exception
695
696

KeyError: Caught KeyError in DataLoader worker process 0.
Original Traceback (most recent call last):
File "/usr/local/lib/python3.10/dist-packages/torch/utils/data/_utils/worker.py", line 308, in _worker_loop
data = fetcher.fetch(index)
File "/usr/local/lib/python3.10/dist-packages/torch/utils/data/_utils/fetch.py", line 51, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/usr/local/lib/python3.10/dist-packages/torch/utils/data/_utils/fetch.py", line 51, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "", line 104, in getitem
rawdata = self.datainfos[idx]
File "/content/Pose2Seg/datasets/CocoDatasetInfo.py", line 130, in getitem
return self.getitem(idx)
File "/content/Pose2Seg/datasets/CocoDatasetInfo.py", line 216, in getitem
if isinstance(obj['segmentation'], list):
KeyError: 'segmentation'