#python #feature-extraction #transfer-learning
Вопрос:
Я использую предварительно подготовленную модель Alex-Net для извлечения особенностей некоторых изображений в Fc7 (непосредственно перед классификацией). Каждый раз, когда я извлекаю 4096 функций из каждого изображения, они меняются … почему?
Комментарии:
1. Просто для того, чтобы уточнить, что эти разные функции принадлежат одному и тому же изображению в вашем случае? Если да, то поделитесь своим кодом.
2.Привет, @AbhishekPrajapat, да! для одного и того же изображения функции меняются при каждом запуске. Я замечаю это в своем коде:
layer = 'Fc7' alexNet = models.alexnet(pretrained=True) new_classifier = nn.Sequential(*list(alexNet.classifier.children())[:-1]) alexNet.classifier = new_classifierмне не хватало AlexNet.eval (), так что я думаю, что в этом причина