hujiecpp / ISTR

ISTR: End-to-End Instance Segmentation with Transformers (https://arxiv.org/abs/2105.00637)
202 stars 28 forks source link

onnx export #25

Open Tord-Zhang opened 1 year ago

Tord-Zhang commented 1 year ago

Hi, is there any solution for onnx export? I tried to export onnx, but the exported onnx cannot be used for inference. Error information: "onnxruntime.capi.onnxruntime_pybind11_state.Fail: [ONNXRuntimeError] : 1 : FAIL : Load model from onnx/model.onnx failed:Node (If_4653) Op (If) [TypeInferenceError] Mismatched tensor element type: source=9 target=2"