ValueError:DepthwiseConv2D要求步幅属性包含4个值,但得到:3

2024-04-26 08:22:23 发布

您现在位置:Python中文网/ 问答频道 /正文

将重新训练的(coco权重和自定义数据集)EfficientSet-d0 pytorch实现导出到onnx后,我尝试将onnx保存的模型导入Tensorflow,并在尝试运行预测或导出模型时出现以下错误:

ValueError:DepthwiseConv2D要求stride属性包含4个值,但得到的结果是:“{node depthwise}}=DepthwiseConv2Native[T=DT_FLOAT,data_format=“NHWC”,dillations=[1,1,1,1,1],explicit_paddings=[],padding=“VALID”,strides=[1,1,1]](转置_4,重塑_2)和输入形状:[?,?,?,?],[3,3,32,1]

代码:

onnx_model = onnx.load(PATH + 'model.onnx')
onnx.checker.check_model(onnx_model)
tf_rep = prepare(onnx_model)
tf_rep.export_graph("d0_s256_b32_ep400_TF.pb")

onnx_model = onnx.load(PATH + 'model.onnx')
onnx.checker.check_model(onnx_model)
tf_rep = prepare(onnx_model)
tf_rep.run(Variable(torch.randn(32,3,256,256, dtype=torch.float32)))