sklearn - Как создать правильные метки с несколькими значениями - PullRequest
0 голосов
/ 26 ноября 2018

В моем курсе мы должны реализовать алгоритм распознавания работающего светофора, используя python с общими модулями (numpy, sklearn и т. Д.)

В моем наборе данных simple_train есть 209 изображений, где каждый из нихна картинках нет ни одного, ни одного светофора.Для упрощения я изменил свой алгоритм загрузки, чтобы он просто возвращал один светофор (если он есть или несколько).

Мой набор данных Independet (массивы изображений) имеет форму (209, 720, 1280, 3)и я заметил, что мне нужно преобразовать его в 2-мерный массив - поэтому мой массив наборов функций имеет форму (209, 2764800).

Теперь есть сложная часть, в которой я застрял.Каждый светофор состоит из четырех координат (x_min, x_max, y_min и y_max).Это означает, что для первых трех изображений мой ярлык будет выглядеть следующим образом:

array([None, None, (610, 351, 615, 358), ... ])

На первых двух снимках светофоров нет, а на третьем - светофор с заданными границами.

Вызов функции .fit MLPClassifier Я получаю следующее сообщение об ошибке:

ValueError: Неизвестный тип метки: (array ([None, None, (610, 351, 615), 358), ...])

Как мне изменить мой ярлык, чтобы это заработало?

...