微软开源的深度学习模型转换工具MMdnn
MMdnn简介
在工业街和学术界中，开发者或者研究人员可以选择多种框架来构建模型，每种框架有自己特有的网络结构定义以及模型保存格式，这种框架之间的鸿沟阻碍了不同框架模型之间的操作。MMdnn是微软开源的用于不同深度学习框架(Caffe, Keras, MXNet, Tensorflow, CNTK, PyTorch Onnx和CoreML)的模型之间互相转换的工具，通过模型的中间表示来完成不同框架模型之间的转换。其代码被托管在微软的Github中：https://github.com/Microsoft/MMdnn
支持的框架
目前MMdnn支持的深度学习框架有：
Caffe
Microsoft Cognitive Toolkit (CNTK)
CoreML
Keras
MXNet
ONNX
PyTorch
TensorFlow (实验性的)
DarkNet (只在源码中有, 实验性的)
已测试的模型转换
MMdnn在一些ImageNet模型上测试的结果：
MMdnn的安装
MMdnn可以使用pip直接安装：
pip install mmdnn
使用下面的方法可以安装版本的MMdnn:
pip install -U git+https://github.com/Microsoft/MMdnn.git@master
用MMdnn进行模型的转换
MMdnn可以使用pip直接安装：
模型的转换只需要一行命令，例如下面的命令会下载TensorFlow的ResNet V2 152模型，并使用MMdnn将其转换为PyTorch模型：
mmdownload -f tensorflow -n resnet_v2_152 -o ./
mmconvert -sf tensorflow -in imagenet_resnet_v2_152.ckpt.meta -iw imagenet_resnet_v2_152.ckpt --dstNodeName MMdnn_Output -df pytorch -om tf_resnet_to_pth.pth
用MMdnn进行模型可视化
下载预训练的模型：
mmdownload -f keras -n inception_v3
将预训练模型转换为中间表示：
mmtoir -f keras -w imagenet_inception_v3.h5 -o keras_inception_v3
打开MMdnn模型可视化工具http://mmdnn.eastasia.cloudapp.azure.com:8080/并选择文件keras_inception_v3.json：
官方示例教程
Keras "inception V3"转换为CNTK模型：https://github.com/Microsoft/MMdnn/blob/master/docs/keras2cntk.md
TensorFlow slim模型ResNet V2 152转换为PyTorch模型：https://github.com/Microsoft/MMdnn/blob/master/docs/tf2pytorch.md
Mxnet模型LResNet50E-IR转换为TensorFlow模型：https://github.com/Microsoft/MMdnn/issues/85
声明：文章收集于网络，如有侵权，请联系小编及时处理，谢谢！
时间:2018-10-29 23:18 来源: 转发量:次
声明：本站部分作品是由网友自主投稿和发布、编辑整理上传，对此类作品本站仅提供交流平台，转载的目的在于传递更多信息及用于网络分享，并不代表本站赞同其观点和对其真实性负责，不为其版权负责。如果您发现网站上有侵犯您的知识产权的作品，请与我们取得联系，我们会及时修改或删除。
相关文章：
- [机器学习]使用PyTorch进行知识蒸馏的代码示例
- [机器学习]改进机器学习：知识图谱如何为数据赋予更深层的意义？
- [机器学习]简述机器学习加速器的五种类型
- [机器学习]基于深度学习的低光照增强技术解析
- [机器学习]SPSS的 Process v4.1 插件下载和安装
- [机器学习]PROCESS的安装、卸载、更新方法【如何安装、卸载、更新SPSS自定义对话框】
- [机器学习]十个用于 AutoML 的 GitHub 存储库
- [机器学习]为机器学习模型设置最佳阈值：0.5是二元分类的最佳阈值吗
- [机器学习]部署您的第一个机器学习API
- [机器学习]深度学习的工作原理：窥视驱动今日AI的神经网络的内部
相关推荐：
网友评论：
最新文章
热门文章
