专栏算法工具链onnx转bin模型出错

onnx转bin模型出错

Tristin2023-09-08
40
2
1.芯片型号：J3
2.问题定位：模型转换
4.问题具体描述：错误log：ValueError: cannot reshape array of size 1354752 into shape (3,320,320)。

分析：我模型训练图像尺寸为320x320，在执行 bash 02_preprocess.sh 时采用的校正图片也是320x320的，为什么在转换是会是1354752？按照我的训练图片和校正图片，应该是1x3x320x320=307200才对。麻烦horzion解答一下。

算法工具链
评论1
0/1000
  • huayue
    Lv.1
    试试不用处理过程，直接带入符合格式的数据呢，一行一行输出preprocess数据格式应该就知道问题在哪里了吧？
    2023-09-09
    0
    1
    • Tristin回复huayue:

      谢谢！已解决。

      2023-09-11
      0