keras-onnx
keras-onnx copied to clipboard
keras custom Capsule Layers to onnx
Hey! I'm working on a model that uses custom capsule layers. I used custom_operations to convert it to onnx but not achieved. Anyone here converted multiple custom layers of keras to onnx? If so how? Thanks!
This is the API: keras2onnx.set_converter(<layer.type>, <custom_conversion_function>)
<custom_conversion_function> can be reference here: https://github.com/onnx/keras-onnx/tree/c4efae793dabd301bac23adea2230d5fe30482c7/keras2onnx/ke2onnx, a lot of example how layer was converted.