-
Notifications
You must be signed in to change notification settings - Fork 136
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Ski2DPose dataset #50
Comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
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'
The text was updated successfully, but these errors were encountered: