摘要:是微軟開源的用于不同深度學(xué)習(xí)框架和的模型之間互相轉(zhuǎn)換的工具,通過模型的中間表示來完成不同框架模型之間的轉(zhuǎn)換。
MMdnn簡(jiǎn)介
在工業(yè)街和學(xué)術(shù)界中,開發(fā)者或者研究人員可以選擇多種深度學(xué)習(xí)框架來構(gòu)建模型,每種框架有自己特有的網(wǎng)絡(luò)結(jié)構(gòu)定義以及模型保存格式,這種框架之間的鴻溝阻礙了不同框架模型之間的操作。MMdnn是微軟開源的用于不同深度學(xué)習(xí)框架(Caffe, Keras, MXNet, Tensorflow, CNTK, PyTorch Onnx和CoreML)的模型之間互相轉(zhuǎn)換的工具,通過模型的中間表示來完成不同框架模型之間的轉(zhuǎn)換。其代碼被托管在微軟的Github中:https://github.com/Microsoft/MMdnn
支持的框架
目前MMdnn支持的深度學(xué)習(xí)框架有:
Caffe
Microsoft Cognitive Toolkit (CNTK)
CoreML
Keras
MXNet
ONNX?
PyTorch
TensorFlow (實(shí)驗(yàn)性的)?
DarkNet (只在源碼中有, 實(shí)驗(yàn)性的)
已測(cè)試的模型轉(zhuǎn)換
MMdnn在一些ImageNet模型上測(cè)試的結(jié)果:
MMdnn的安裝
MMdnn可以使用pip直接安裝:
pip install mmdnn
使用下面的方法可以安裝版本的MMdnn:
pip install -U git+https://github.com/Microsoft/MMdnn.git@master
用MMdnn進(jìn)行模型的轉(zhuǎn)換
MMdnn可以使用pip直接安裝:
模型的轉(zhuǎn)換只需要一行命令,例如下面的命令會(huì)下載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進(jìn)行模型可視化
下載預(yù)訓(xùn)練的模型:
mmdownload -f keras -n inception_v3
將預(yù)訓(xùn)練模型轉(zhuǎn)換為中間表示:
mmtoir -f keras -w imagenet_inception_v3.h5 -o keras_inception_v3
打開MMdnn模型可視化工具h(yuǎn)ttp://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)絡(luò),如有侵權(quán),請(qǐng)聯(lián)系小編及時(shí)處理,謝謝!歡迎加入本站公開興趣群商業(yè)智能與數(shù)據(jù)分析群
興趣范圍包括各種讓數(shù)據(jù)產(chǎn)生價(jià)值的辦法,實(shí)際應(yīng)用案例分享與討論,分析工具,ETL工具,數(shù)據(jù)倉(cāng)庫(kù),數(shù)據(jù)挖掘工具,報(bào)表系統(tǒng)等全方位知識(shí)
QQ群:81035754
文章版權(quán)歸作者所有,未經(jīng)允許請(qǐng)勿轉(zhuǎn)載,若此文章存在違規(guī)行為,您可以聯(lián)系管理員刪除。
轉(zhuǎn)載請(qǐng)注明本文地址:http://systransis.cn/yun/4815.html
摘要:完整版地址我們的想法是創(chuàng)建一個(gè)深度學(xué)習(xí)框架的羅塞塔石碑假設(shè)你很了解某個(gè)深度學(xué)習(xí)框架,你就可以幫助別人使用任何框架。我們的目標(biāo)是創(chuàng)建深度學(xué)習(xí)框架的羅塞塔石碑,使數(shù)據(jù)科學(xué)家能夠在不同框架之間輕松運(yùn)用專業(yè)知識(shí)。 repo 1.0 完整版 GitHub 地址:https://github.com/ilkarman/DeepLearningFrameworks我們的想法是創(chuàng)建一個(gè)深度學(xué)習(xí)框架的羅塞塔石...
摘要:微軟在大會(huì)上興奮地宣布了的預(yù)覽版,這是一種跨平臺(tái)的開源機(jī)器學(xué)習(xí)框架。核心組件作為的一部分而發(fā)布今天的代碼倉(cāng)庫(kù)包含用于模型訓(xùn)練和使用的,以及許多常見的機(jī)器學(xué)習(xí)任務(wù)如回歸和分類所需要的各種轉(zhuǎn)換和學(xué)習(xí)器。 微軟在Build 2018大會(huì)上興奮地宣布了ML.NET的預(yù)覽版,這是一種跨平臺(tái)的開源機(jī)器學(xué)習(xí)框架。ML.NET將讓廣大.NET開發(fā)人員可以開發(fā)自己的模型,并且將自定義的機(jī)器學(xué)習(xí)融入到其應(yīng)用程序...
摘要:但年月,宣布將在年終止的開發(fā)和維護(hù)。性能并非最優(yōu),為何如此受歡迎粉絲團(tuán)在過去的幾年里,出現(xiàn)了不同的開源深度學(xué)習(xí)框架,就屬于其中典型,由谷歌開發(fā)和支持,自然引發(fā)了很大的關(guān)注。 Keras作者Fran?ois Chollet剛剛在Twitter貼出一張圖片,是近三個(gè)月來arXiv上提到的深度學(xué)習(xí)開源框架排行:TensorFlow排名第一,這個(gè)或許并不出意外,Keras排名第二,隨后是Caffe、...
摘要:微軟今天開源微軟認(rèn)知工具包的升級(jí)版本,升級(jí)版。他同時(shí)也是微軟認(rèn)知工具包的關(guān)鍵架構(gòu)師。微軟的認(rèn)知工具包代表了微軟與英偉達(dá)緊密的合作,共同推動(dòng)深度學(xué)習(xí)社區(qū)的進(jìn)步,英偉達(dá)加速計(jì)算部門總經(jīng)理說。微軟認(rèn)知工具包可在多個(gè)上運(yùn)行,其中包括的產(chǎn)品。 微軟今天開源微軟認(rèn)知工具包(Microsoft Cognitive Toolkit)的升級(jí)版本,CNTK 升級(jí)版。本次升級(jí)較大的亮點(diǎn)在于增加了 Python 綁...
摘要:近日,與微軟聯(lián)合推出了開放式神經(jīng)網(wǎng)絡(luò)交換格式,它是一個(gè)表征深度學(xué)習(xí)模型的標(biāo)準(zhǔn),可實(shí)現(xiàn)模型在不同框架之間的遷移。例如,在中,條件句通常是對(duì)輸入張量的大小或維度上的計(jì)算。 近日,F(xiàn)acebook 與微軟聯(lián)合推出了開放式神經(jīng)網(wǎng)絡(luò)交換(ONNX)格式,它是一個(gè)表征深度學(xué)習(xí)模型的標(biāo)準(zhǔn),可實(shí)現(xiàn)模型在不同框架之間的遷移。ONNX 是構(gòu)建開源生態(tài)環(huán)境的第一步,供人工智能開發(fā)者輕松選擇并組合較先進(jìn)的工具。開發(fā)...
閱讀 2867·2021-07-30 15:30
閱讀 562·2019-08-30 15:55
閱讀 1631·2019-08-26 17:04
閱讀 642·2019-08-26 11:36
閱讀 2082·2019-08-26 10:58
閱讀 3562·2019-08-23 14:34
閱讀 1566·2019-08-22 18:48
閱讀 2533·2019-08-21 17:51