Как настроить AlexNet для 3 классов вместо 1000? - PullRequest
0 голосов
/ 28 апреля 2018

Я использую AlexNet из здесь

Данные там имеют 1000 классов, поэтому они имеют весовые коэффициенты в соответствии с этим. Как заставить его работать для прогнозирования значений для моих данных, которые имеют 3 класса?

Я знаю, что должен изменить вес, но я не знаю как.

Ответы [ 2 ]

0 голосов
/ 28 апреля 2018

Вам нужно

добавить еще один слой поверх предварительно обученной сети

. Это будет ваш

выходной слой

.

Эти выходные данные слоя классов 1000 будут входными данными этого слоя, и он даст ваши 3 класса в качестве выходных данных.

После этого тренируйте эту новую сеть с вашими изображениями

0 голосов
/ 28 апреля 2018

Вам просто нужно установить num_classes = 3, что уменьшит количество классов вывода как для тензора вывода модели, так и для отдельно определенного заполнителя y.

Количество весов, то есть параметров, будет соответствующим образом изменено при вызове model = AlexNet(....

...