Conversion of MODNet .onnx model

aadhithya / onnx-typecast

Script to typecast ONNX model parameters from INT64 to INT32.

MIT License

91 stars 19 forks source link

Conversion of MODNet .onnx model #6

Open lee101 opened 10 months ago

lee101 commented 10 months ago

Thanks for sharing this script, i tried it on MODNet. (background removal open source codebase)

I had an error running it, it worked but when trying to do inference i get an error due to the ReShape operator

INVALID_GRAPH : Load model from modnetpf16simint32.onnx failed:This is an invalid model. Type Error: Type 'tensor(int32)' of input parameter (480) of operator (Reshape) in node (Reshape_177) is invalid.

aadhithya commented 9 months ago

can you share the link to the model repo/weights?