关于一些知名深度学习模型的转换

因为最近的项目需求,需要我将Tensorflow模型、caffe模型和Mxnet模型转换成darknet模型,因此做个记录。

一些github上面的模型转换汇总:https://github.com/ysh329/deep-learning-model-convertor

从网上现有的一些模型转换方法可以看出,能转成darknet的模型里面目前仅有caffe,

因此对于Tensorflow的转换我们可以使用

Tensorflow---->caffe MMdnn    nn_tools

caffe---->darknet  pytorch-caffe-darknet-convert

 

对于Mxnet,我们可以使用

mxnet---->caffe   MMdnn   Mxnet2Caffe

caffe---->darknet  pytorch-caffe-darknet-convert

对于caffe2,我们可以使用

 

caffe2---->pytorch ONNX

pytorch--->caffe   pytorch-caffe-darknet-convert

caffe---->darknet  pytorch-caffe-darknet-convert

 

posted @ 2018-07-09 14:04  chengjue  阅读(4127)  评论(1编辑  收藏  举报