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

Я пытаюсь изменить вес модели кофе, которая является частью филиала кафе под названием 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?


caffe pycaffe
person mcExchange    schedule 04.10.2016    source источник

Ответы (1)


arrow_upward
2
arrow_downward

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

person mcExchange    schedule 04.10.2016