Как настроить AlexNet для 3 классов вместо 1000?

Question

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

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

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

Answer 1

Вам нужно

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

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

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

.

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

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

Answer 2

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

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