Skip to content

added example and tests for exporting keras models #32

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 7 commits into
base: master
Choose a base branch
from

Conversation

kmader
Copy link

@kmader kmader commented May 15, 2017

I added tests for simple CNN-based Keras models (#18) as well as the pretrained models which then cover the popular image classification models (ResNet, VGG, Inception), currently all of the tests pass but there is a hard coded list of UNSUPPORTED_LAYERS which are tolerated failures and show the corresponding missing operation on the tfdeploy side (#31)

Model 1: use_upsample could not be serialized unknown operation: ResizeNearestNeighbor
Model 5: use_lstm could not be serialized unknown operation: TensorArrayReadV3
Model 6: use_dropout could not be serialized unknown operation: Merge
Model 7: use_locallyconnected could not be serialized unknown operation: BatchMatMul
Model 8: use_repeatvec could not be serialized unknown operation: StridedSlice
Model 9: use_conv2dtrans could not be serialized unknown operation: Conv2DBackpropInput
Model 10: use_bn could not be serialized unknown operation: Merge

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant