Дисбаланс и небольшой набор данных для классификации изображений

Question

У меня есть два набора данных изображений для классификации:

• Повреждение шоколадной упаковки из 27 изображений
• Неповрежденная шоколадная упаковка из 161 изображений

Я должен написать классификатор Python, чтобы различать и сигнализацию.

Как лучше всего решить мою проблему? CNN с трансферным обучением или решение среднего значения?

Answer 1

Похоже, у вас действительно мало данных. Для меня трансферное обучение - лучший способ справиться с этим. Тем не менее, попробуйте найти больше данных или использовать увеличение данных (вы можете прочитать this ).

Кроме того, если вы ищете переводное обучение с CNN, я думаю, этот урок может вам помочь.