关于一些知名深度学习模型的转换
因为最近的项目需求,需要我将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