1 引言

目前地平線浮點(diǎn)轉(zhuǎn)換工具鏈需要依據(jù) yaml 配置文件實(shí)現(xiàn)模型轉(zhuǎn)換，而某些參數(shù)如果配置有誤將會(huì)導(dǎo)致最終模型的性能/精度不理想，甚至直接轉(zhuǎn)換失敗，本文將針對(duì) yaml 文件中的相關(guān)配置項(xiàng)進(jìn)行說明，作為工具鏈文檔的額外補(bǔ)充。

Note：隨著工具鏈版本迭代，yaml 參數(shù)可能會(huì)有細(xì)微調(diào)整，本文以 v1.9.4 版本 MobileNet-v1為例進(jìn)行說明。

2 概述

整個(gè) yaml 文件包含 4 個(gè)必選參數(shù)組（model_parameters、input_parameters、calibration_parameters、compiler_parameters）和 1 個(gè)可選參數(shù)組（custom_op），每個(gè)參數(shù)組下也區(qū)分必選和可選參數(shù)（可選參數(shù)默認(rèn)隱藏），具體要求和填寫方式請(qǐng)參考工具鏈文檔。其中以下為幾個(gè)填寫注意點(diǎn)：

• 當(dāng)模型為多輸入模型時(shí), 請(qǐng)將可選參數(shù) input_name, input_shape顯式寫出, 以免造成參數(shù)對(duì)應(yīng)順序上的錯(cuò)誤；
• 當(dāng) input_type_rt 配置為 nv12 時(shí)，模型H/W維度不支持奇數(shù)輸入大小，例如不支持 1x255x255x3。

3 參數(shù)詳解

本小節(jié)針對(duì)各部分參數(shù)進(jìn)行分類說明，主要包括 性能影響參數(shù)、精度影響參數(shù)、自定義算子參數(shù) 三類。關(guān)于參數(shù)的具體描述或取值范圍，可參考對(duì)應(yīng)的工具鏈文檔。

3.1 性能影響參數(shù)

關(guān)于配置文件中與性能相關(guān)參數(shù)總結(jié)（包含未在示例中暴露的高階參數(shù)）如下表所示：

圖1 Layer Details信息

3.2 精度影響參數(shù)

關(guān)于配置文件中與精度相關(guān)參數(shù)總結(jié)（包含未暴露的參數(shù)）如下表所示：

3.3 自定義算子參數(shù)

此部分參數(shù)針對(duì)您有自定義算子行為的需求進(jìn)行配置，目前我們已開放部分CPU算子自定義，更多關(guān)于如何自定義算子請(qǐng)參考文章自定義算子這一小節(jié)，相關(guān)參數(shù)配置說明如下：

本文轉(zhuǎn)載自地平線開發(fā)者社區(qū)：https://developer.horizon.ai
原作者：顏值即正義