В настоящее время я тренирую модель для обнаружения изображений и хочу знать, сколько изображений мне нужно для каждого класса, нужно ли мне иметь одинаковые номера каждого объекта.
Пожалуйста, мне нужен совет.
Я использую Tensorflow и модель Yolo v2.
Спасибо,
Вам нужно столько, сколько вы можете получить, но определенно порядка десятков тысяч, по крайней мере, если вы тренируете сеть с нуля (для обучения YOLOv2 есть предварительно обученные веса - http://host.robots.ox.ac.uk/pascal/VOC/).
Лучше всего иметь сбалансированные классы, то есть количество изображений для каждого класса должно быть близким, так легче тренироваться.
Почему вы сами тренируете сеть? Разве вы не можете использовать некоторые предварительно обученные модели, удалить слои FC и вставить свои собственные классы? Таким образом, это намного быстрее, и вам не нужно так много изображений.
