Найти тему
Devpractice.ru

Создание классификатора изображений с использованием Transfer Learning и PyTorch

Pytorch + Transfer Learning
Pytorch + Transfer Learning

Технология Transfer Learning позволяет применять для решения конкретной задачи нейросеть, обученную на других данных (сходных по тематике). Технически это выглядит так: к уже натренированной сети вы добавляете несколько слоев и обучаете их для решения более конкретной задачи. Подробно об этом подходе с примерами на PyTorch написано в статье:

Image Classification with Transfer Learning and PyTorch