多通道图片预处理BUG

[sml2h3]

多通道图片预处理BUG

[lazy-ape]

对于彩色的图片确实是这个问题，我改了一下，没有报错了，但是识别不出来内容，不知道是模型的问题，还是改的问题。
这个错误的原因应该是图片要 reshape 后传入模型，因为图片的shape 是 h, w, channel, 模型要求的是 channel, h, w。

我的代码改动如下，可以作为参考：

在 __init__.py 的 classification 方法，第 1635 行增加以下代码

 if self.__channel == 3:
            image = np.asarray(image).astype(np.float32)
            old_shape = image.shape
            image = image.reshape(old_shape[0], old_shape[3], old_shape[1], old_shape[2])
            ort_inputs = {'input1': image}
  else:
            ort_inputs = {'input1': np.array([image]).astype(np.float32)}

[ljyh223]

请问问题解决了吗，我也遇到了

[ztshun]

遇到了同样的问题，没解决

[ztshun]

对于彩色的图片确实是这个问题，我改了一下，没有报错了，但是识别不出来内容，不知道是模型的问题，还是改的问题。 这个错误的原因应该是图片要 reshape 后传入模型，因为图片的shape 是 h, w, channel, 模型要求的是 channel, h, w。

我的代码改动如下，可以作为参考：

在 __init__.py 的 classification 方法，第 1635 行增加以下代码

 if self.__channel == 3:
            image = np.asarray(image).astype(np.float32)
            old_shape = image.shape
            image = image.reshape(old_shape[0], old_shape[3], old_shape[1], old_shape[2])
            ort_inputs = {'input1': image}
  else:
            ort_inputs = {'input1': np.array([image]).astype(np.float32)}

试了，没报错，但是不能识别

[ztshun]

INVALID_ARGUMENT : Invalid rank for input: input1 Got: 5 Expected: 4 Please fix either the inputs or the model

请问解决了吗

[Qinlane]

感谢
在 ort_outs = self.__ort_session.run(None, ort_inputs) 前面加上你的判断就可以了