算能科技
直播中

王静

7年用户 1543经验值
私信 关注
[问答]

bmneto转换模型时报错误split_conv_float: Assertion \"valid\" failed是什么原因?

我们在项目中要把BackgroundMattingV2模型的pytorch格式转成bm格式,转换时先把pytorch模型转成onnx模型,再用bmneto转bm模型,转bm模型时报错误:split_conv_float: Assertion "valid" failed。请问这个报错是什么原因,怎么解决?
SDK版本:22.12.01
模型转换命令:
python3 -m bmneto --model ./bm.onnx
        --input_names "src,bgr"
        --shapes "[[1,3,1080,1920],[1,3,1080,1920]]"
        --target BM1684X
        --outdir bmodel/fp32
        --output_names "pha,fgr,err,hid"
        --cmp False
报错截图:

更多回帖

发帖
×
20
完善资料,
赚取积分