【模型推理】Tengine 模型轉(zhuǎn)換及量化

歡迎關(guān)注我的公眾號 [極智視界]膝昆,回復001獲取Google編程規(guī)范

O_o ? >_< ? o_O ? O_o ? ~_~ ? o_O

本文介紹一下 Tengine 模型轉(zhuǎn)換及量化流程帮寻。

Tengine 同 ncnn 一樣该编,也是優(yōu)秀的端側(cè)推理框架,前面寫過一篇《ncnn 模型轉(zhuǎn)換及量化流程》爷辙,有興趣的同學可以查閱彬坏。

下面開始。

1膝晾、編譯 Tengine Tools

CmakeList.txt 中打開 CONVERT_TOOL 和 QUANT_TOOL 開關(guān):

image

開始編譯:

cd <Tengine-path>
mkdir build

cd build

# 開始編譯
camke ..
make -j32
make install

編譯完會在 ./build/install/bin/ 目錄下生成 convert_toolquant_tool栓始,如下:

image

2、Tengine 模型轉(zhuǎn)換

用如下命令看一下 convert_tool 需要什么樣的傳參:

cd ./build/install/bin

./convert_tool -h
image

如上血当,前面為輸入幻赚,后面為輸出。

Tengine 提供的模型轉(zhuǎn)換工具前端支持的十分豐富臊旭,基本通吃了國外坯屿、國內(nèi)主流的深度學習框架。

前端為 Caffe:

./convert_tool -f caffe -p ./mobilenet.prototxt -m ./mobilenet.caffemodel -o ./mobilenet.tmfile

前端為 Onnx:

./convert_tool -f onnx -m ./mobilenet.onnx -o ./mobilenet.tmfile</pre>

前端為 Mxnet:

./convert_tool -f mxnet -p ./mobilenet.params -m ./mobilenet.json -o ./mobilenet.tmfile

前端為 Darknet:

./convert_tool -f darknet -p ./yolov3.weights -m ./yolov3.cfg -o yolov3.tmfile

前端為 TensorFlow:

./convert_tool -f tensorflow -m mobielenet_v1_1.0_224_frozen.pb -o mobilenet.tmfile

前端為 TFLite:

./convert_tool -f tflite -m mobielenet.tflite -o mobilenet.tmfile

前端為 MegEngine:

./convert_tool -f megengine -m mobilenet.pkl -o mobilenet.tmfile

前端為 OneFlow:

./convert_tool -f oneflow -p mobilenet.prototxt -m mobilenet/ -o mobilenet.tmfile

前端為 ncnn:

./convert_tool -f ncnn -p mobilenet.param -m mobilenet.bin -o mobilenet.tmfile

3巍扛、Tengine 量化推理

可以看到 Tengine 提供的量化工具有三個:quant_tool_int8领跛、quant_tool_uint8quant_tool_uint8_perchannel撤奸,分別是對稱量化吠昭、非對稱量化、非對稱逐通道量化胧瓜,里面的量化策略類似矢棚,這里挑 quant_tool_int8說一下。

先安裝一下依賴庫:

apt update
apt install libopencv-dev

用如下命令查看量化工具傳參:

./quant_tool_int8 -h
image

可以看到傳參還是比較豐富的府喳。

前端輸入為 fp32 tmfile蒲肋,執(zhí)行量化示例:

./quant_tool_int8  -m ./mobilenet_fp32.tmfile -i ./dataset -o ./mobilenet_int8.tmfile -g 3,224,224 -w 104.007,116.669,122.679 -s 0.017,0.017,0.017

執(zhí)行成功會輸出如下日志:

image

并生成如下 int8 模型文件:

image

然后就可以拿去執(zhí)行推理了。

以上分享了 Tengine 模型轉(zhuǎn)換及量化方法钝满,希望我的分享能對你的學習有一點幫助兜粘。

【公眾號傳送】
【模型推理】Tengine 模型轉(zhuǎn)換及量化

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市弯蚜,隨后出現(xiàn)的幾起案子孔轴,更是在濱河造成了極大的恐慌,老刑警劉巖碎捺,帶你破解...
    沈念sama閱讀 218,036評論 6 506
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件路鹰,死亡現(xiàn)場離奇詭異,居然都是意外死亡收厨,警方通過查閱死者的電腦和手機晋柱,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 93,046評論 3 395
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來诵叁,“玉大人雁竞,你說我怎么就攤上這事±栊荩” “怎么了浓领?”我有些...
    開封第一講書人閱讀 164,411評論 0 354
  • 文/不壞的土叔 我叫張陵玉凯,是天一觀的道長。 經(jīng)常有香客問我联贩,道長漫仆,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 58,622評論 1 293
  • 正文 為了忘掉前任泪幌,我火速辦了婚禮盲厌,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘祸泪。我一直安慰自己吗浩,他們只是感情好,可當我...
    茶點故事閱讀 67,661評論 6 392
  • 文/花漫 我一把揭開白布没隘。 她就那樣靜靜地躺著懂扼,像睡著了一般。 火紅的嫁衣襯著肌膚如雪右蒲。 梳的紋絲不亂的頭發(fā)上阀湿,一...
    開封第一講書人閱讀 51,521評論 1 304
  • 那天,我揣著相機與錄音瑰妄,去河邊找鬼陷嘴。 笑死,一個胖子當著我的面吹牛间坐,可吹牛的內(nèi)容都是我干的灾挨。 我是一名探鬼主播,決...
    沈念sama閱讀 40,288評論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼竹宋,長吁一口氣:“原來是場噩夢啊……” “哼劳澄!你這毒婦竟也來了?” 一聲冷哼從身側(cè)響起逝撬,我...
    開封第一講書人閱讀 39,200評論 0 276
  • 序言:老撾萬榮一對情侶失蹤浴骂,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后宪潮,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 45,644評論 1 314
  • 正文 獨居荒郊野嶺守林人離奇死亡趣苏,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 37,837評論 3 336
  • 正文 我和宋清朗相戀三年狡相,在試婚紗的時候發(fā)現(xiàn)自己被綠了。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片食磕。...
    茶點故事閱讀 39,953評論 1 348
  • 序言:一個原本活蹦亂跳的男人離奇死亡尽棕,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出彬伦,到底是詐尸還是另有隱情滔悉,我是刑警寧澤伊诵,帶...
    沈念sama閱讀 35,673評論 5 346
  • 正文 年R本政府宣布,位于F島的核電站回官,受9級特大地震影響曹宴,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜歉提,卻給世界環(huán)境...
    茶點故事閱讀 41,281評論 3 329
  • 文/蒙蒙 一笛坦、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧苔巨,春花似錦版扩、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,889評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至悼尾,卻和暖如春柿扣,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背诀豁。 一陣腳步聲響...
    開封第一講書人閱讀 33,011評論 1 269
  • 我被黑心中介騙來泰國打工窄刘, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人舷胜。 一個月前我還...
    沈念sama閱讀 48,119評論 3 370
  • 正文 我出身青樓娩践,卻偏偏與公主長得像,于是被迫代替她去往敵國和親烹骨。 傳聞我的和親對象是個殘疾皇子翻伺,可洞房花燭夜當晚...
    茶點故事閱讀 44,901評論 2 355

推薦閱讀更多精彩內(nèi)容