keras-onnx icon indicating copy to clipboard operation
keras-onnx copied to clipboard

keras custom Capsule Layers to onnx

Open prnvjb opened this issue 6 years ago • 1 comments

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!

prnvjb avatar Nov 15 '19 09:11 prnvjb

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.

wenbingl avatar Nov 15 '19 22:11 wenbingl