Tensorflow - переобучение для распознавания этнической принадлежности и цвета волос

Question

Я новичок в тензорном потоке и начальной модели. Я нашел следующий учебник онлайн (https://www.tensorflow.org/versions/master/how_tos/image_retraining/) и хотел проверить это на моем собственном проекте.

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

По какой-то причине я не могу получить более 65% уровня точности. Я попытался увеличить количество тренировок и количество изображений.

Возможно, оригинальная модель не подходит для этой модели?

Может ли кто-нибудь указать мне в хорошем смысле, что я могу сделать, чтобы улучшить результаты?

С уважением, П.

Answer 1

Вы получаете точность 65% на поезде или на тестовом наборе? Если он находится в наборе поездов, вы, вероятно, что-то не так с кодом. Если он установлен на тестовом наборе, вы действительно используете неправильную модель. Модель Inception - очень большая модель, и только 850 изображений для каждой категории не дадут вам хорошей общей модели. Он просто «запомнит» эти 850 изображений. (подумайте о том, чтобы запомнить ответ на каждый вопрос теста, а не учиться на тест)

Возможно, вы можете попробовать сначала создать более простую, меньшую модель и посмотреть, насколько хорошо эта модель учится!