我们在项目中要把BackgroundMat
tingV2模型的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
报错截图: