search

cv516Buaa / tph-yolov5

GNU General Public License v3.0
716 stars 170 forks source link

export导出onnx模型后，使用onnx模型进行推理报错 #63

Open kfchandsome opened 11 months ago

kfchandsome commented 11 months ago

onnxruntime.capi.onnxruntime_pybind11_state.Fail: [ONNXRuntimeError] : 1 : FAIL : Node (If_726) Op (If) [TypeInferenceError] Graph attribute inferencing failed: Node (Squeeze_728) Op (Squeeze) [ShapeInferenceError] Dimension of input 3 must be 1 instead of 128