Grundsätzlich benutze ich Caffe für Deep Learning. Jetzt habe ich ein Netzwerk für mein eigenes Dataset verfeinert und mein Modell im hdf5-Format gespeichert, welches die .h5 Erweiterung ist.Wie konvertiert man das hdf5-Format in das Caffemodel-Format in caffe?
Jetzt möchte ich Bilder mit Matcaffe klassifizieren. matcaffe versteht nur .caffemodel Format für trainiertes Modell.
Gibt es eine Möglichkeit, hdf5-Datei in Caffemodel zu konvertieren?
Sie können das bevorzugte Format für die Datei caffemodel in Ihrem solver.prototxt festlegen. Einfach einstellen
snapshot_format: BINARYPROTO
Weitere Informationen finden Sie unter caffe.proto.
Alternativ können Sie Python-Schnittstelle (which is by far better than the matlab interface for caffe, IMHO) verwenden:
import caffe
net = caffe.Net('/path/to/deploy.prototxt', '/path/to/caffemodel.h5', caffe.TEST)
net.save('/path/to/just.caffemodel')
Shai danke für deine Antwort, eigentlich habe ich anfangs nicht gewusst, dass es verschiedene Snapshot-Formate gibt, also habe ich im HDF5-Format gespeichert. Ich lief 1akh Iterationen, so dass es mir nicht möglich war, es noch einmal auszuführen. –
Aber ich fand einen Weg dafür und es hat für mich funktioniert. –
@RafayZiaMir können Sie die von Ihnen gefundene Problemumgehung näher erläutern? würdest du es als Antwort posten? – Shai
, wie Sie das hdf5 Modell gespeichert haben während der Feinabstimmung? – Shai