微軟開源的深度學習模型轉(zhuǎn)換工具MMdnn

高璐發(fā)布于2019-04-25 18:29 / 3753人閱讀

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

MMdnn簡介

在工業(yè)街和學術界中，開發(fā)者或者研究人員可以選擇多種深度學習框架來構建模型，每種框架有自己特有的網(wǎng)絡結構定義以及模型保存格式，這種框架之間的鴻溝阻礙了不同框架模型之間的操作。MMdnn是微軟開源的用于不同深度學習框架(Caffe, Keras, MXNet, Tensorflow, CNTK, PyTorch Onnx和CoreML)的模型之間互相轉(zhuǎn)換的工具，通過模型的中間表示來完成不同框架模型之間的轉(zhuǎn)換。其代碼被托管在微軟的Github中：https://github.com/Microsoft/MMdnn

支持的框架

目前MMdnn支持的深度學習框架有：

Caffe

Microsoft Cognitive Toolkit (CNTK)

CoreML

Keras

MXNet

ONNX?

PyTorch

TensorFlow (實驗性的)?

DarkNet (只在源碼中有, 實驗性的)

已測試的模型轉(zhuǎn)換

MMdnn在一些ImageNet模型上測試的結果：

MMdnn的安裝

MMdnn可以使用pip直接安裝：

pip install mmdnn

使用下面的方法可以安裝版本的MMdnn:

pip install -U git+https://github.com/Microsoft/MMdnn.git@master

用MMdnn進行模型的轉(zhuǎn)換

MMdnn可以使用pip直接安裝：

模型的轉(zhuǎn)換只需要一行命令，例如下面的命令會下載TensorFlow的ResNet V2 152模型，并使用MMdnn將其轉(zhuǎn)換為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

將預訓練模型轉(zhuǎn)換為中間表示：

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"轉(zhuǎn)換為CNTK模型：https://github.com/Microsoft/MMdnn/blob/master/docs/keras2cntk.md

TensorFlow slim模型ResNet V2 152轉(zhuǎn)換為PyTorch模型：https://github.com/Microsoft/MMdnn/blob/master/docs/tf2pytorch.md

Mxnet模型LResNet50E-IR轉(zhuǎn)換為TensorFlow模型：https://github.com/Microsoft/MMdnn/issues/85

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

商業(yè)智能與數(shù)據(jù)分析群

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

QQ群：81035754

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

文章版權歸作者所有，未經(jīng)允許請勿轉(zhuǎn)載,若此文章存在違規(guī)行為，您可以聯(lián)系管理員刪除。

轉(zhuǎn)載請注明本文地址：http://www.ezyhdfw.cn/yun/4815.html

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

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

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

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

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

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

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

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

ymyang 2019-04-25 18:07 評論0 收藏0
Facebook聯(lián)合微軟推出神經(jīng)網(wǎng)絡交換格式ONNX

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

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