Настроенный набор данных MPII python3 класс для Pytorch 1.3 - PullRequest
0 голосов
/ 26 января 2020

Я новичок в этом виде обработки изображений. И я хочу построить набор данных MPII , описанный здесь класс для обучения сети Pytorch на нем. Я склонен работать с Python 3.6 и pytorch 1.4

Проблема главным образом в том, что аннотация набора данных MPII (скачать здесь) в форме .mat, с которой я не очень знаком с. Есть также некоторые реализации, которые я нашел на Github, такие как глубокое высокое разрешение net, а также этот набор данных Microsoft MPII для python. Последний фактически принят во многих реализациях в наборе данных MPII, но он слишком стар. Он построен с pytorch == 0.4.0 и python2, создавая множество проблем с экологическими конфликтами.

Теперь я хочу создать простой набор данных pytorch, который может автоматически извлекать объединенные позиции в каталоге 'annotation /' и комбинировать его с указанным c изображением в 'image /' каталог. Я надеюсь, что возвращаемое значение - это просто словарь с этими двумя элементами, это все, что мне нужно.

Я также заметил, что многие исследователи склонны конвертировать исходный файл аннотации .mat в формат. json. , Я думаю, что это облегчит работу, но есть некоторые проблемы, когда я пытаюсь это сделать.

Python 3.6.10 |Anaconda, Inc.| (default, Jan  7 2020, 21:14:29)
[GCC 7.3.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import json
>>> from scipy.io import loadmat
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/home/billy/miniconda3/envs/datasci/lib/python3.6/json/__init__.py", line 231, in dumps
    return _default_encoder.encode(obj)
  File "/home/billy/miniconda3/envs/datasci/lib/python3.6/json/encoder.py", line 199, in encode
    chunks = self.iterencode(o, _one_shot=True)
  File "/home/billy/miniconda3/envs/datasci/lib/python3.6/json/encoder.py", line 257, in iterencode
    return _iterencode(o, 0)
  File "/home/billy/miniconda3/envs/datasci/lib/python3.6/json/encoder.py", line 180, in default
  File "/home/billy/miniconda3/envs/datasci/lib/python3.6/json/encoder.py", line 180, in default
    o.__class__.__name__)
TypeError: Object of type 'bytes' is not JSON serializable

Я также видел этот вопрос на этом сайте, но Я не уверен, что понимаю, что не так ... Я очень признателен за любую помощь !!

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...