Как можно сохранить подмножество предварительно обученной модели кафе?

Я работаю над предварительно обученной моделью кафе (на питоне), которая имеет 3 слоя. Я хочу разложить эту модель кафе и создать новую модель, аналогичную первому слою этой модели. Например:

Исходные данные модели Caffe -> conv1_1 -> conv1_2 -> conv2_1 -> conv2_2 -> conv3_1 -> conv3_2

Новые данные модели Caffe -> conv1_1 -> conv1_2

Кто-нибудь может мне помочь?


caffe pre-trained-model
person Ali    schedule 15.08.2016    source источник

Ответы (1)


arrow_upward
0
arrow_downward

Python предоставляет данные внутри файла .caffemodel. Доступ к нему можно получить как к массиву. Например,

net = caffe.Net('path/to/conv.prototxt', 'path/to/conv.caffemodel', caffe.TEST)
W = net.params['con_1'][0].data[...]
b = net.params['con_1'][1].data[...]

Вы можете скопировать эти данные в новый файл и сохранить его как файл .caffemodel. Взгляните на это и это.

person malreddysid    schedule 16.08.2016