1 引言

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

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

2 概述

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

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

3 參數(shù)詳解

本小節(jié)針對各部分參數(shù)進行分類說明，主要包括 性能影響參數(shù)、精度影響參數(shù)、自定義算子參數(shù) 三類。關(guān)于參數(shù)的具體描述或取值范圍，可參考對應(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ù)針對您有自定義算子行為的需求進行配置，目前我們已開放部分CPU算子自定義，更多關(guān)于如何自定義算子請參考文章自定義算子這一小節(jié)，相關(guān)參數(shù)配置說明如下：

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