华为计算微信公众号
昇腾AI开发者公众号
华为计算微博
华为计算今日头条
由自己的pytorch网络模型生成om模型后在Atalas200DK上运行发现输入与om模型输入不匹配,但是已经对输入图片做了相应的resize,报错如下。
感觉可能是由于数据类型的问题,于是想在代码中修改输入的数据类型为模型需要的float32,但是发现通过videocapture读取之后的图片是AclImage类的对象,无法将其转成nparray类型再转换数据类型。或者怎么改变AclImage类输入图片的数据类型呢?
本帖最后由 匿名用户 于 2024/06/27 16:36:12 编辑
我要发帖子
由自己的pytorch网络模型生成om模型后在Atalas200DK上运行发现输入与om模型输入不匹配,但是已经对输入图片做了相应的resize,报错如下。
感觉可能是由于数据类型的问题,于是想在代码中修改输入的数据类型为模型需要的float32,但是发现通过videocapture读取之后的图片是AclImage类的对象,无法将其转成nparray类型再转换数据类型。或者怎么改变AclImage类输入图片的数据类型呢?