Significantly improved conversion success rate for models containing `Einsum` - Githubissues

PINTO0309 / onnx2tf

Self-Created Tools to convert ONNX files (NCHW) to TensorFlow/TFLite/Keras format (NHWC). The purpose of this tool is to solve the massive Transpose extrapolation problem in onnx-tensorflow (onnx-tf). I don't need a Star, but give me a pull request.

MIT License

712 stars 73 forks source link

Significantly improved conversion success rate for models containing `Einsum` #497

Closed PINTO0309 closed 1 year ago

PINTO0309 commented 1 year ago

1. Content and background

Einsum
- Significantly improved conversion success rate for models containing Einsum.
- Special techniques significantly optimize model structures for which model optimization with onnx.shape_inference and onnxsim fails.
- However, relief is provided only when there are no undefined dimensions in the input tensor.
- https://huggingface.co/SmilingWolf/wd-v1-4-moat-tagger-v2
- https://s3.ap-northeast-2.wasabisys.com/temp-models/onnx2tf_497/wd-v1-4-moat-tagger-v2.onnx

2. Summary of corrections

3. Before/After (If there is an operating log that can be used as a reference)

4. Issue number (only if there is a related issue)