微軟開源的深度學習模型轉換工具MMdnn

高璐發布于2019-04-25 18:29 / 3646人閱讀

摘要：是微軟開源的用于不同深度學習框架和的模型之間互相轉換的工具，通過模型的中間表示來完成不同框架模型之間的轉換。

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

聲明：文章收集于網絡，如有侵權，請聯系小編及時處理，謝謝！歡迎加入本站公開興趣群

商業智能與數據分析群

興趣范圍包括各種讓數據產生價值的辦法，實際應用案例分享與討論，分析工具，ETL工具，數據倉庫，數據挖掘工具，報表系統等全方位知識

QQ群：81035754

GPU云服務器云服務器學習深度學習用什么工具深度學習的深度深度學習的學習學習深度學習的

文章版權歸作者所有，未經允許請勿轉載,若此文章存在違規行為，您可以聯系管理員刪除。

轉載請注明本文地址：http://specialneedsforspecialkids.com/yun/4815.html

現代羅塞塔石碑：微軟提出深度學習框架的通用語言

摘要：完整版地址我們的想法是創建一個深度學習框架的羅塞塔石碑假設你很了解某個深度學習框架，你就可以幫助別人使用任何框架。我們的目標是創建深度學習框架的羅塞塔石碑，使數據科學家能夠在不同框架之間輕松運用專業知識。 repo 1.0 完整版 GitHub 地址：https://github.com/ilkarman/DeepLearningFrameworks我們的想法是創建一個深度學習框架的羅塞塔石...

ckllj 2019-04-25 18:26 評論0 收藏0
微軟開源 ML.NET：一款跨平臺、成熟的機器學習框架

摘要：微軟在大會上興奮地宣布了的預覽版，這是一種跨平臺的開源機器學習框架。核心組件作為的一部分而發布今天的代碼倉庫包含用于模型訓練和使用的，以及許多常見的機器學習任務如回歸和分類所需要的各種轉換和學習器。微軟在Build 2018大會上興奮地宣布了ML.NET的預覽版，這是一種跨平臺的開源機器學習框架。ML.NET將讓廣大.NET開發人員可以開發自己的模型，并且將自定義的機器學習融入到其應用程序...

linkin 2019-04-25 18:26 評論0 收藏0
ArXiv最受歡迎開源深度學習框架榜單：TensorFlow第一

摘要：但年月，宣布將在年終止的開發和維護。性能并非最優，為何如此受歡迎粉絲團在過去的幾年里，出現了不同的開源深度學習框架，就屬于其中典型，由谷歌開發和支持，自然引發了很大的關注。 Keras作者Fran?ois Chollet剛剛在Twitter貼出一張圖片，是近三個月來arXiv上提到的深度學習開源框架排行：TensorFlow排名第一，這個或許并不出意外，Keras排名第二，隨后是Caffe、...

trilever 2019-04-25 18:24 評論0 收藏0
微軟開源深度學習認知工具包：增加 Python 綁定，支持增強學習

摘要：微軟今天開源微軟認知工具包的升級版本，升級版。他同時也是微軟認知工具包的關鍵架構師。微軟的認知工具包代表了微軟與英偉達緊密的合作，共同推動深度學習社區的進步，英偉達加速計算部門總經理說。微軟認知工具包可在多個上運行，其中包括的產品。微軟今天開源微軟認知工具包（Microsoft Cognitive Toolkit）的升級版本，CNTK 升級版。本次升級較大的亮點在于增加了 Python 綁...

ymyang 2019-04-25 18:07 評論0 收藏0
Facebook聯合微軟推出神經網絡交換格式ONNX

摘要：近日，與微軟聯合推出了開放式神經網絡交換格式，它是一個表征深度學習模型的標準，可實現模型在不同框架之間的遷移。例如，在中，條件句通常是對輸入張量的大小或維度上的計算。近日，Facebook 與微軟聯合推出了開放式神經網絡交換（ONNX）格式，它是一個表征深度學習模型的標準，可實現模型在不同框架之間的遷移。ONNX 是構建開源生態環境的第一步，供人工智能開發者輕松選擇并組合較先進的工具。開發...

CrazyCodes 2019-04-25 18:17 評論0 收藏0