专栏算法工具链模型输入非图像时如何编译

模型输入非图像时如何编译

已解决
TJ2024-07-24
81
3

1. 芯片型号:J5

2. 天工开物开发包 OpenExplorer 版本:v1.1.68_20231014

3. 问题定位:PTQ模型编译问题

4. 问题具体描述:如果模型的输入不为图像,仅为一维向量,那么在使用hb_mapper makertbin编译时,应该如何设置模型编译的yaml配置文件,具体体现在input_parameters参数应该如何设置,似乎默认选项都是以图片输入为主的,是否有在不改变模型结构的情况下,可以将一维向量输入的模型进行编译呢?

# 输入信息参数组

input_parameters:

 # 原始浮点模型的输入节点名称

 input_name: "data"

 # 原始浮点模型的输入数据格式(数量/顺序与input_name一致)

 input_type_train: 'bgr'

 # 原始浮点模型的输入数据排布(数量/顺序与input_name一致)

 input_layout_train: 'NCHW'

 # 原始浮点模型的输入数据尺寸

 input_shape: '1x3x224x224'

 # 网络实际执行时,输入给网络的batch_size, 默认值为1

 input_batch: 1

 # 在模型中添加的输入数据预处理方法

 norm_type: 'data_mean_and_scale'

 # 预处理方法的图像减去的均值, 如果是通道均值,value之间必须用空格分隔

 mean_value: '103.94 116.78 123.68'

 # 预处理方法的图像缩放比例,如果是通道缩放比例,value之间必须用空格分隔

 scale_value: '0.017'

 # 转换后混合异构模型需要适配的输入数据格式(数量/顺序与input_name一致)

 input_type_rt: 'yuv444'

 # 输入数据格式的特殊制式

 input_space_and_range: 'regular'

 # 转换后混合异构模型需要适配的输入数据排布(数量/顺序与input_name一致),若input_type_rt配置为nv12,则此处参数不需要配置

 input_layout_rt: 'NHWC'

算法工具链
征程5
评论3
0/1000
  • kotei左文亮
    Lv.3

    输入是一维的数据,那是模型啊 训练时候的预处理方式是什么样的啊? 可以设置成1*1*1*n试一试

    2024-07-25
    0
    0
  • Huanghui
    Lv.5

    你好,这个问题你可以:

    1. 提供模型给我们,我们内部通过多中配置方式尝试看看。

    2. 你自己将input_type_train和input_type_rt设置成feathmap试试!

    2024-07-25
    0
    0
  • kotei左文亮
    Lv.3

    鉴于您长时间未回复,此问题就先关闭了。如还有疑问,可再发帖求助。

    2024-08-05
    0
    0