Сетевая хирургия на заказной модели кафе

Question

Я пытаюсь изменить вес кофейной модели, которая является частью кофейной ветви Deep Lab. Несмотря на то, что есть руководство по сетевой хирургии, когда я пытаюсь сделать то же самое с моей пользовательской кофейной моделью, ядро ​​python всегда умирает в следующей строке:

# 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

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