search

onnx / keras-onnx

Convert tf.keras/Keras models to ONNX
Apache License 2.0
381 stars 109 forks source link

keras custom Capsule Layers to onnx #295

Open prnvjb opened 5 years ago

prnvjb commented 5 years ago

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!

wenbingl commented 5 years ago

This is the API: keras2onnx.set_converter(, )

can be reference here: https://github.com/onnx/keras-onnx/tree/c4efae793dabd301bac23adea2230d5fe30482c7/keras2onnx/ke2onnx, a lot of example how layer was converted.