Особенности Caffe2 extract fc7

Question

недавно я пытался использовать Convolutional Neural Network (CNN) для извлечения особенностей изображений.Модель, которую я использовал - alexnet, фреймворк - Caffe2 с Python 3.6.4.

Проблема в том, что я хочу извлечь объекты слоя FC7, который должен иметь размер 4096.

Как мне решить эту проблему?

Answer 1

Вы должны использовать функцию FetchBlob(blob_name) рабочего пространства.В вашем случае (при условии, что имя выхода FC7 - 'fc7'):

blob = workspace.FetchBlob('fc7')

Теперь вы можете легко напечатать форму вывода:

print(blob.shape)