TFLite模型終極開(kāi)盲盒來(lái)啦!
大家好!前兩期的內容,不知道大家是不是都已經(jīng)體驗過(guò)了呢~(TFLite模型的開(kāi)盲盒體驗 |TFLite模型的開(kāi)盲盒體驗之編輯模型 )
本文引用地址:http://dyxdggzs.com/article/202405/458557.htm上期小編帶著(zhù)大家從源碼開(kāi)始編譯了一個(gè)叫做flatc的小工具??梢杂脕?lái)實(shí)現對于TFLite模型格式的解析與生成。然后也帶著(zhù)大家,手動(dòng)編輯了以下json文件,為模型添加了一個(gè)新的輸出節點(diǎn):
相信大家一定不想僅僅局限于手動(dòng)編輯吧?那小編就滿(mǎn)足大家的好奇心,繼續給大家看看如何利用python腳本進(jìn)行自動(dòng)化編輯,用代碼的方式,編譯TFLite模型。
在開(kāi)始前,先給大家介紹一下何謂JSON格式:
JSON(JavaScript Object Notation)是一種輕量級的數據交換格式,基于JavaScriptJSON(JavaScript Object Notation)是一種輕量級的數據交換格式,基于JavaScript編程語(yǔ)言的一個(gè)子集。它的主要優(yōu)點(diǎn)包括:
◆ 數據格式簡(jiǎn)單,易于讀寫(xiě)
◆ 格式已壓縮,占用帶寬小
◆ 易于解析,客戶(hù)端JavaScript可通過(guò)eval ()進(jìn)行JSON數據的讀取
◆ 支持多種語(yǔ)言,包括ActionScript, C, C#, ColdFusion, Java, JavaScript, Perl, PHP, Python, Ruby等服務(wù)器端語(yǔ)言,便于服務(wù)器端的解析
◆ PHP-JSON和JSON-PHP已經(jīng)存在PHP領(lǐng)域,偏于PHP序列化后的程序直接調用,PHP服務(wù)器端的對象、數組等能直接生成JSON格式,便于客戶(hù)端的訪(fǎng)問(wèn)提取
◆ JSON格式可直接為服務(wù)器端代碼使用,極大簡(jiǎn)化了服務(wù)器端和客戶(hù)端的代碼開(kāi)發(fā)量,且完成任務(wù)不變,并易于維護
與此同時(shí),JSON也存在一些不足:
◆ 沒(méi)有XML格式推廣和使用的廣泛性,沒(méi)有XML的通用性
◆ 目前在Web Service中推廣還屬于初級階段
好啦!書(shū)歸正傳,我們在使用python進(jìn)行處理時(shí)就需要用到j(luò )son庫,使用方式很直接,同時(shí)為了保證都進(jìn)來(lái)的json文件中的元素不會(huì )被打亂順序,需要用到OrderedDict:
接下來(lái)就是將json文件讀取進(jìn)來(lái):
不過(guò),此時(shí)直接進(jìn)行讀取,會(huì )報錯:
這是因為flatc轉換出來(lái)的json文件不太符合標準,需要進(jìn)行一下修改:
這樣一來(lái),就可以成功打開(kāi)了:
獲取模型輸出:
可以看到果然是我們上次所修改的0和11:
現在我們就可以直接進(jìn)行修改,將輸出變回去,只有一個(gè)0作為輸出:
重新保存json文件:
利用flatc工具導出tflite格式,模型會(huì )保存成cifar10_new.tflite:
使用模型可視化工具將模型打開(kāi):
再檢查下模型輸入輸出參數:
轉換成功!這樣我們就可以通過(guò)編寫(xiě)python腳本更加靈活的對TFLite格式文件進(jìn)行處理。當然實(shí)際上是利用python的json庫對json文件進(jìn)行處理,間接實(shí)現改造TFLite文件的目的。
至此,全部TFLite模型的開(kāi)盲盒體驗就完結了。希望大家發(fā)揮想象,使TFLite模型成為自由可編輯的模型格式!加油!
開(kāi)盲盒第一篇:TFLite模型的開(kāi)盲盒體驗 http://dyxdggzs.com/article/202405/458552.htm
開(kāi)盲盒第二篇:TFLite模型的開(kāi)盲盒體驗之編輯模型 http://dyxdggzs.com/article/202405/458554.htm
評論