Чистая хирургия на заказной модели модели

Question

Я пытаюсь изменить весы 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?

Answer 1

Я нашел его уже. Я буквально удалял каждый пользовательский уровень и особенно адаптировал уровень данных таким образом, чтобы он мог читать все входные изображения и тем самым вычислять размеры ввода.