Можем ли мы обучить предварительно обученную модель новым данным, используя тензорный поток?

Question

У меня есть обученная модель, которая может классифицировать кошку или собаку как файл h5 с именем

catsvsdogs.h5

Теперь я хочу добавить еще одно животное в список. Скажем, я хочу классифицировать кошку / собаку / тигра, используя эту модель, и я не хочу повторять процесс обучения кошки и собаки, вместо этого я хочу передать набор данных о тигре в свою нейронную сеть, а затем он классифицирует собаку или кошку или тигр с тем же файлом. Возможно ли это сделать? Если да, то как?

Answer 1

К сожалению, это невозможно. Конечно, вы можете обучить модель для tiger класса, но вам также понадобятся данные dogs & cats для того, чтобы модель могла классифицировать эти 3 класса.

Если вы загрузите этот (catsvsdogs.h5) предварительно Обученная модель для обучения только в классе tiger. Вы должны будете установить выходной слой с классом 1, поэтому в момент прогнозирования он будет только предсказывать, тигр он или нет.

Переподготовка обученной модели только в новом классе - все еще вопрос исследования.

Answer 2

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