输入大小与om模型输入大小不匹配
收藏回复举报
输入大小与om模型输入大小不匹配
t('forum.solved') 已解决
新人帖
发表于2023-11-14 01:24:04
0 查看

由自己的pytorch网络模型生成om模型后在Atalas200DK上运行发现输入与om模型输入不匹配,但是已经对输入图片做了相应的resize,报错如下。

cke_7789.png

感觉可能是由于数据类型的问题,于是想在代码中修改输入的数据类型为模型需要的float32,但是发现通过videocapture读取之后的图片是AclImage类的对象,无法将其转成nparray类型再转换数据类型。或者怎么改变AclImage类输入图片的数据类型呢?

本帖最后由 匿名用户2024/06/27 16:36:12 编辑

我要发帖子