专栏算法工具链hbm模型图像输入能否设置为rgb格式数据?

hbm模型图像输入能否设置为rgb格式数据?

chengang2023-08-01
39
3
1.芯片型号:J5
2.天工开物开发包OpenExplorer版本:J5_OE_1.1.57
3.问题定位:hbm模型能否输入为rgb格式数据
4.问题具体描述

你好,看手册ptq量化生成的bin模型可以设置板端输入图像为rgb格式。请问qat方式生成的hbm模型是不是也可以设置上板模型输入为rgb格式?如果是的话,是要修改编译选项input_source吗?

算法工具链
评论2
0/1000
  • 颜值即正义
    Lv.2

    您好,如果想要hbm模型的输入格式为rgb,需要您在编译的时候设置input_source为ddr。不过需要注意的是,在准备板端数据的时候,需要把输入数据手动做归一化、量化;若存在硬件对齐的话,还需要手动做padding;最后把输入数据转为bin文件。

    2023-08-01
    0
    1
    • chengang回复颜值即正义:

      好的,非常感谢,解决了我之前的一个疑惑。

      2023-08-01
      0
  • 颜值即正义
    Lv.2

    您好,地平线工具链在持续迭代优化,为了给您提供更好的服务,希望您能抽出3分钟左右的时间,将您在使用工具链期间的感受和建议告诉我们,您的宝贵意见对我们很重要,非常感谢!

    2023-08-01
    0
    0