2016-10-04 3 views
1

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

ответ

2

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

 Смежные вопросы

  • Нет связанных вопросов^_^