Как партия может включать как положительные, так и отрицательные ярлыки? - PullRequest
0 голосов
/ 10 октября 2018

У меня есть проблема классификации пешеходов, которую я хочу решить с помощью VGG-16.Для этого я подготовил обучающие и тестовые наборы.У моего набора поездов 2038 изображений, а у моего тестового набора - 252 изображения.Размер моей партии - 64. Как я могу сказать Keras, что я хочу, чтобы эти 64 изображения включали как положительные, так и отрицательные ярлыки во время тренировки?Я не хочу, чтобы он учился только на положительных или только на отрицательных ярлыках.

1 Ответ

0 голосов
/ 30 января 2019

Если вы перетасуете свои тренировочные данные (что настоятельно рекомендуется), очень малый шанс получить все 64 образца, принадлежащие к одному и тому же классу.Для задачи с двумя классами не должно быть проблем с данными.

Однако, если вы хотите гарантировать сбалансированные учебные партии, вы можете использовать сторонний код, такой как BalancedBatchGenerator .См. учебник для получения дополнительной информации.

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