Я пытаюсь изменить весы caffemodel, которая является частью отделения caffe под названием Deep Lab. Хотя есть учебник по how to do net surgery, когда я пытаюсь сделать то же самое с моим обычаем caffemodel ядра питона всегда умирает на следующей строке:Чистая хирургия на заказной модели модели
# Load the original network and extract the fully connected layers' parameters.
net = caffe.Net('../models/deeplab/train.prototxt',
'../models/deeplab/train.caffemodel',
caffe.TRAIN)
Я думаю его потому, что pycaffe не знает их собственные слои, такие как ImageSegData
, Silence
и SegAccuracy
, поэтому я удалил эти слои из файла prototxt, но все же ядро python продолжает умирать, когда я пытаюсь загрузить сетевую модель. Кто-нибудь знает, как загрузить эти веса в python?