更新时间:2024-04-03 GMT+08:00
分享

模型配置文件

模型配置文件是一个json文件,模型在加载的时候将会根据这个文件自动加载合适的模型并进行输入的预处理。本小节将介绍该json文件的具体结构。

表1 参数说明

参数名

参数类型

是否必填

说明

ModelCfgs

ArrayList<ModelCfg>

模型配置。ModelCfg参数说明请见表2

表2 ModelCfg参数说明

参数名

参数类型

是否必填

说明

DeviceType

String

设备类型。支持的设备类型有:D310_C32、D310_C30、D310_B902、D310_C75、D310_C76、D310_C77、NNIE、 D310_C3X、D310_C7X。其中D310_C3X包含D310_C32、D310_C30、D310_B902;D310_C7X包含D310_C75、 D310_C76、D310_C77。

ModelPath

String

模型路径。模型路径可以是绝对路径,也可以是相对路径,相对路径就是相对于当前模型配置文件的路径。

FrameWorkInput

String

框架输入格式。支持NV21、JPEG。JPEG仅支持DeviceType为D310系列的设备类型。

ModelInput

String

模型输入格式。支持NV21,后续可能会扩展。

NetworkWidth

Integer

模型输入宽度。

NetworkHeight

Integer

模型输入高度。

NetworkChannel

Integer

模型输入通道数。

ScaleMeanEnable

Boolean

是否需要进行输入减均值缩放处理。可以填写的值:true、false。当FrameWorkInput为JPEG时,必填。

Scale

Float

缩放值。当ScaleMeanEnable为true时,必填。

MeanR_Y

Float

R通道均值。当模型的输入只有一个通道时,使用该通道。当ScaleMeanEnable为true时,必填。

MeanG

Float

G通道均值。当ScaleMeanEnable为true时,必填。

MeanB

Float

B通道均值。当ScaleMeanEnable为true时,必填。

相关文档