onnx转bin模型出错

应用开发 算法工具链
1

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解答一下。

2

试试不用处理过程，直接带入符合格式的数据呢，一行一行输出preprocess数据格式应该就知道问题在哪里了吧？

3

谢谢！已解决。