The project will be updated continuously ......
Pull requests are welcome!
Deep Learning Model Convertors
Note: This is not one convertor for all frameworks, but a collection of different converters. Because github is an open source platform, I hope we can help each other here, gather everyone's strength.
Because of these different frameworks, the awesome convertors of deep learning models for different frameworks occur. It should be noted that I did not test all the converters, so I could not guarantee that each was available. But I also hope this convertor collection may help you!
The sheet below is a overview of all convertors in github (not only contain official provided and more are user-self implementations). I just make a little work to collect these convertors. Also, hope everyone can support this project to help more people who're also crazy because of various frameworks.
Brief Intro of Convertors
Open Neural Network Exchange
General framework for converting between all kinds of neural networks
ONNX is an effort to unify converters for neural networks in order to bring some sanity to the NN world. Released by Facebook and Microsoft. More info here.
MMdnn
MMdnn is a set of tools to help users inter-operate among different deep learning frameworks. E.g. model conversion and visualization. Convert models between Caffe, Keras, MXNet, Tensorflow, CNTK and PyTorch.
MXNet convertor
Convert to MXNet model.
mdering/CoreMLZoo: A few models converted from caffe to CoreMLs format
A few deep learning models converted from various formats to CoreMLs format.
Models currently available:
- SqueezeNet
- VGG19
Please feel free to create a pull request with additional models.
mxnet/tools/caffe_converter
Key topics covered include the following:
- Converting Caffe trained models to MXNet
- Calling Caffe operators in MXNet More concretly, please refer to mxnet/tools/caffe_converter and How to | Convert from Caffe to MXNet.
nicklhy/ResNet_caffe2mxnet
This is a tool to convert the deep-residual-networks from caffe model to mxnet model. The weights are directly copied from caffe network blobs.
Caffe convertor
Convert to Caffe model.
pytorch2caffe
Convert PyTorch model to Caffemodel.
cypw/MXNet2Caffe
Convert MXNet model to Caffe model.
wranglerwong/Mxnet2Caffe: Convert MXNet model to Caffe model
Convert MXNet model to Caffe model.
kuangliu/mocha
Convert torch model to/from caffe model easily.
facebook/fb-caffe-exts/torch2caffe
Some handy utility libraries and tools for the Caffe deep learning framework, which has ** A library for converting pre-trained Torch models to the equivalent Caffe models.**
pytorch-caffe-darknet-convert
Convert between pytorch, caffe and darknet models. Caffe darknet models can be load directly by pytorch.
Teaonly/trans-torch
Translating Torch model to other framework such as Caffe, MxNet ...
e-lab/th2caffe
A torch-nn to caffe converter for specific layers.
hahnyuan/nn_tools
a neural network convertor for models among caffe tensorflow keras
Caffe2 convertor
Convert to Caffe2 model.
onnx-caffe2
Convert PyTorch to Caffe2 (making it especially easy to deploy on mobile devices)
CNTK convertor
Convert to CNTK model.
crosstalkcaffe/CaffeConverter
The tool will help you convert trained models from Caffe to CNTK.
Convert trained models: giving a model script and its weights file, export to CNTK model.
crosstalk
crosstalk is from CNTK contrib.
Theano/Lasagne convertor
Convert to Theano/Lasagne model.
an-kumar/caffe-theano-conversion
This is part of a project for CS231N at Stanford University, written by Ankit Kumar, Mathematics major, Class of 2015
This is a repository that allows you to convert pretrained caffe models into models in Lasagne, a thin wrapper around Theano. You can also convert a caffe model's architecture to an equivalent one in Lasagne. You do not need caffe installed to use this module.
Currently, the following caffe layers are supported:
* Convolution
* LRN
* Pooling
* Inner Product
* Relu
* Softmax
kencoken/caffe-model-convert
Convert models from Caffe to Theano format.
piergiaj/caffe-to-theano
Convert a Caffe Model to a Theano Model. This currently works on AlexNet, but should work for any Caffe model that only includes layers that have been impemented.
Neon convertor
Convert to Neon model.
NervanaSystems/caffe2neon
Tools to convert Caffe models to neon's serialization format.
This repo contains tools to convert Caffe models into a format compatible with the neon deep learning library. The main script, "decaffeinate.py", takes as input a caffe model definition file and the corresponding model weights file and returns a neon serialized model file. This output file can be used to instantiate the neon Model object, which will generate a model in neon that should replicate the behavior of the Caffe model.
PyTorch convertor
Convert to PyTorch model.
ruotianluo/pytorch-resnet
Convert resnet trained in caffe to pytorch model.
clcarwin/convert_torch_to_pytorch
Convert torch t7 model to pytorch model and source.
vzhong/chainer2pytorch
chainer2pytorch implements conversions from Chainer modules to PyTorch modules, setting parameters of each modules such that one can port over models on a module basis.
pytorch-caffe-darknet-convert
Convert between pytorch, caffe and darknet models. Caffe darknet models can be load directly by pytorch.
nn-transfer
Convert between Keras and PyTorch models.
Torch convertor
Convert to Torch model.
kmatzen/googlenet-caffe2torch
Converts bvlc_googlenet.caffemodel to a Torch nn model.
Want to use the pre-trained GoogLeNet from the BVLC Model Zoo in Torch? Do you not want to use Caffe as an additional dependency inside Torch? Use these two scripts to build the network definition in Torch and copy the learned weights from the Caffe model.
kuangliu/mocha
Convert torch model to/from caffe model easily.
Keras convertor
Convert to Keras model.
qxcv/caffe2keras
Note: This converter has been adapted from code in Marc Bolaños fork of Caffe. See acks for code provenance.
This is intended to serve as a conversion module for Caffe models to Keras models.
Please, be aware that this module is not regularly maintained. Thus, some layers or parameter definitions introduced in newer versions of either Keras or Caffe might not be compatible with the converter. Pull requests welcome!
hahnyuan/nn_tools
a neural network convertor for models among caffe tensorflow keras
masterhou/caffe2keras
Caffe to Keras converter, From https://github.com/MarcBS/keras. This is intended to serve as a conversion module for Caffe models to Keras Functional API models.
MarcBS/keras
Keras' fork with several new functionalities. Caffe2Keras converter, multimodal layers, etc. https://github.com/MarcBS/keras
This fork of Keras offers the following contributions:
Caffe to Keras conversion module Layer-specific learning rates New layers for multimodal data Contact email: marc.bolanos@ub.edu
GitHub page: https://github.com/MarcBS
MarcBS/keras is compatible with: Python 2.7 and Theano only.
OdinLin/caffe2keras
a simple tool to translate caffe model to keras model.
jamescfli/Deep_Learning_Model_Converter
nn-transfer
Convert between Keras and PyTorch models.
pytorch2keras
Convert pytorch models to Keras.
Darknet convertor
Convert to Darknet model.
pytorch-caffe-darknet-convert
Convert between pytorch, caffe and darknet models. Caffe darknet models can be load directly by pytorch.
TensorFlow convertor
Convert to TensorFlow model.
crosstalk
crosstalk is from CNTK.
triagemd/model-converters: Tools for converting Keras models for use with other ML frameworks
Tools for converting Keras models for use with other ML frameworks (coreML, TensorFlow).
hahnyuan/nn_tools
a neural network convertor for models among caffe tensorfloe keras
ethereon/caffe-tensorflow: Caffe models in TensorFlow
Convert Caffe models to TensorFlow.
goranrauker/convert-to-tensorflow
Converts a variety of trained models to a frozen tensorflow protocol buffer file for use with the c++ tensorflow api. C++ code is included for using the frozen models.
thtrieu/darkflow
Translate darknet to tensorflow. Load trained weights, retrain/fine-tune using tensorflow, export constant graph def to mobile devices.
dEcmir/lego_yolo
Tensorflow code to to retrain yolo on a new dataset using weights from darknet
This repository contains experiments of transfer learning using YOLO on a new synthetical LEGO data set ROUGH AND UNDOCUMENTED!
alanswx/keras_to_tensorflow
Convert keras models to tensorflow frozen graph for use on cell phones, etc.
amir-abdi/keras_to_tensorflow
General code to convert a trained keras model into an inference tensorflow model.
leonidk/pytorch-tf
Converting a pretrained pytorch model to tensorflow
pytorch2keras
Convert pytorch models to Tensorflow (via Keras)
Chainer convertor
Convert to Chainer model.
coreML convertor
Convert to coreML model.
Apple: Converting Trained Models to Core ML
Convert trained models created with third-party machine learning tools to the Core ML model format.
If your model is created and trained using a supported third-party machine learning tool, you can use Core ML Tools to convert it to the Core ML model format. Table 1 lists the supported models and third-party tools.
| Model type | Supported models | Supported tools |
|---|---|---|
| Neural networks | Feedforward, convolutional, recurrent | Caffe v1 Keras 1.2.2+ |
| Tree ensembles | Random forests, boosted trees, decision trees | scikit-learn 0.18 XGBoost 0.6 |
| Support vector machines | Scalar regression, multiclass classification | scikit-learn 0.18 LIBSVM 3.22 |
| Generalized linear models | Linear regression, logistic regression | scikit-learn 0.18 |
| Feature engineering | Sparse vectorization, dense vectorization, categorical processing | scikit-learn 0.18 |
| Pipeline models | Sequentially chained models | scikit-learn 0.18 |
mxnet/tools/mxnet-to-coreml
Convert MXNet models into Apple CoreML format.
This tool helps convert MXNet models into Apple CoreML format which can then be run on Apple devices.
prisma-ai/torch2coreml: Torch7 -> CoreML
This tool helps convert Torch7 models into Apple CoreML format which can then be run on Apple devices.
woffle/torch2ios: Torch7 Library - Convert NN Models To iOS Format
Torch7 Library - Convert NN Models To iOS Format.
Small lib to serialise Torch7 Networks for iOS. Supported Layers include Fully Connected, Pooling and Convolution Layers at present. The library stores the weights & biases (if any) for each layer necesarry for inference on iOS devices.
Bulochkin/keras_models: Keras models with python-based convertor to provide embedding in IOS platform
Keras models with python-based convertor to provide embedding in IOS platform.
triagemd/model-converters: Tools for converting Keras models for use with other ML frameworks
Tools for converting Keras models for use with other ML frameworks (coreML, TensorFlow).