lFatality / tensorflow2caffe

Convert a model from TensorFlow to Caffe.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

tensorflow2caffe

Convert a model from TensorFlow to Caffe.

The code has been created during this video series:
Part 1 - Creating the architectures
Part 2 - Exporting the parameters
Part 3 - Adapting and comparing

In the videos, the creation of the code has been commented so if you want to get more information about the code you can get it there.

The code uses tflearn, not native Tensorflow.
If you use native Tensorflow, some alterations are necessary (e.g. the export of the parameters).
To understand how to convert succesfully, studying the code might help you.

If you want to convert your own model, start with the export_parameters.py file to get the weights and biases of your model (make sure to change the .model and the layer array and use your own architecture).
Then recreate your architecture in a .prototxt file and use the create_caffemodel.py file to convert your weights and biases to the Caffe format (make sure to change the file so that it fits your network).
Now you should have your .prototxt and .caffemodel in addition to your TensorFlow architecture and .model file.
If you want you can compare the outputs of both networks using the test_network files.

About

Convert a model from TensorFlow to Caffe.

License:MIT License


Languages

Language:Python 100.0%