J'utilise keras pour obtenir un fichier modèle pour .hdf5, comment le convertir en .caffemodel?

Question

J'ai utilisé keras former un modèle, mais je dois prendre caffe à prédire, mais je ne veux pas reconvertir le modèle, alors je veux convertir le fichier .HDF5 à .caffemodel

Answer 1

nn_tools supporte la conversion de/à plusieurs cadres NN, y compris de keras à caffe. Il ne supporte actuellement que les couches conv et fc, mais c'est un bon début: voir keras_to_caffe.py.