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(https://github.com/pytorch/pytorch/issues/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'