Netto-Operation auf einem benutzerdefinierten caffe Modell

Question

Ich versuche, die Gewichte eines caffemodel zu ändern, die Teil eines caffe-Zweig genannt Tief Lab ist. Zwar gibt es ein Tutorial auf how to do net surgery, wenn ich versuche, das gleiche mit meinem benutzerdefinierten caffemodel des Python-Kernel zu tun stirbt immer auf der folgende Zeile:

# Load the original network and extract the fully connected layers' parameters. 
net = caffe.Net('../models/deeplab/train.prototxt', 
       '../models/deeplab/train.caffemodel', 
       caffe.TRAIN) 

Ich denke, sein weil pycaffe wissen nicht, ihre individuellen Schichten wie ImageSegData, Silence und SegAccuracy so entfernten ich diese Schichten aus der prototxt Datei, aber immer noch der python-Kernel hält auf zu sterben, wenn ich versuche, das Netzwerkmodell zu laden. Kann jemand diese Gewichte in Python laden?

Answer 1

ich es schon gefunden. Ich musste buchstäblich jede benutzerdefinierte Ebene entfernen und die Datenschicht speziell so anpassen, dass sie alle Eingabebilder lesen und dadurch die Eingabemaße berechnen konnte.