专栏算法工具链图像分割模型转换过程中报错

图像分割模型转换过程中报错

已解决
zhengsiyu2023-04-10
45
12

请问我在使用mobileunet训练模型时，转换模型过程中遇到如下问题，但是模型转换也完成了，在部署到板子上进行使用时运行结果与预期不符。我需要怎么做才能解决这个问题？输入图像大小为1*512*512*3。

模型训练时Reshape操作如下：

模型的一些信息如下：

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

    你好，第一个图中的log提示不影响模型转换哈。

    关于转换后模型上板精度有问题，建议您先参考一下这篇文章来解决：https://developer.horizon.ai/forumDetail/71036815603174578

    2023-04-10
    0
    10
    • zhengsiyu回复颜值即正义:

      您好我在docker中转换完成后进行推理报错：模型输入大小1*512*512，得到的时1204*2048，与模型输入要求的512*512不一致，脚本中没看出来需要修改哪里分辨率设置。

      2023-04-10
      0
    • 颜值即正义回复zhengsiyu:

      你好，请问你使用的OE版本是多少呀？

      可以配置input_shape参数：

      2023-04-10
      0
    • zhengsiyu回复颜值即正义:

      现在使用的是1.10.7，yaml文件中input_shape已经修改为与模型一致的大小了。

      2023-04-10
      0
    • 颜值即正义回复zhengsiyu:

      1.10.7是很早之前的版本了，X3派的板子，建议切换成OE2.5.2哈，获取链接：https://developer.horizon.ai/forumDetail/136488103547258769

      2023-04-10
      0
    • 颜值即正义回复颜值即正义:

      使用最新版之后，如果还有这个问题，建议将onnx模型/config/校准数据压缩一下，通过百度网盘链接提供一下，我来复现一下你的问题哈

      2023-04-10
      0
    • zhengsiyu回复颜值即正义:

      链接：https://pan.baidu.com/s/1RuS47FNVawolyOoOF72J3g?pwd=j10w

      提取码：j10w

      2023-04-10
      1
    • 颜值即正义回复zhengsiyu:

      你好，我这边在OE2.5.2没有复现你的问题好哈：

      提示一下：板端nv12输入时，它已经不再具有layout的概念了，需要准备nv12对应格式的数据

      2023-04-10
      0
    • 颜值即正义回复zhengsiyu:

      还有就是，文件中提供的校准数据准备的尺寸不太对

      2023-04-10
      0
    • zhengsiyu回复颜值即正义:

      我的校准数据就是训练模型使用的数据512*512的，不对的话我需要修改成多大的呢?板端nv12输入我是调用例子中的bgr2nv12_opencv(image)这个函数，应该没有问题吧?

      2023-04-10
      0
    • 颜值即正义回复zhengsiyu:

      校准数据准备这块，确实是准备成512x512哈，是不是float32或者uint8在使用的时候不太对呢，可以看一下这篇文章来比对哈：https://developer.horizon.ai/forumDetail/118363912788935513，还有就是，建议您更新一下OE包版本哈

      关于板端nv12的输入，建议参考OE包中的转换示例，看看它的nv12数据是如何准备的

      2023-04-11
      0
  • 颜值即正义
    Lv.2
    2023-04-24
    0
    0