专栏算法工具链Mask rcnn求助

Mask rcnn求助

已解决
shaun2021-03-31
122
5

使用工具链转换maskrcnn模型遇到第一个问题（模型来自onnx官网），报错如下图。

模型对输入要求如下图，请问在使用工具链check模型前如何按自己需求设定好数据格式呢 01_checker.sh里面并没有找到数据处理相关函数。env.conf也没有找到相关说明

模型连接：https://github.com/onnx/models/tree/master/vision/object_detection_segmentation/mask-rcnn

算法工具链
评论2
0/1000
  • shaun
    Lv.1

    问题1解决，模型输入默认名称为‘image’，所以在01_checker.sh里面加了一行代码，如下图所示

    但出现了第二个问题，有遇到过的给点建议?

    2021-03-31
    0
    3
    • 芯链情报局回复shaun:

      你好，我们只支持 4维算子，而且 XJ3 芯片上暂时不支持多 batch，所以需要是 1x3x800x800 的输入

      2021-04-01
      0
    • shaun回复芯链情报局:

      您好，请问01_checker.sh具体流程是怎么样的呢，

      模型时对数据有要求吗，目前这些sample里面好像只要求input_name/input_shape.

      下图中的calibrate model这个过程是调用了哪些内容呢？如果调用了其它配置，这些配置文件在哪里进行查看呢？（01_checker.sh与后面的文件（02、03...）存在相互调用关系吗？）

      期待解答

      2021-04-01
      0
    • admin回复shaun:

      大佬，需要多看啊可能shell脚本，只是针对模型的优化，现在是你的模型输入是3*512*512，但是板子支持的模型是1*3*512*512，所以校验失败

      2021-04-01
      0
  • shaun
    Lv.1

    模型输入要求，不是Nx3xHxW

    2021-03-31
    0
    0