Skip to content
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

Unknown layer: SpatialPyramidPooling in predict stage #17

Open
ahmetmelihbasbug opened this issue Nov 25, 2017 · 1 comment
Open

Unknown layer: SpatialPyramidPooling in predict stage #17

ahmetmelihbasbug opened this issue Nov 25, 2017 · 1 comment

Comments

@ahmetmelihbasbug
Copy link

Hello. I tried to import SPP in my model with RNN. I build it and my model.fit function is working. System learned. But in System testing stage, which i call in different function after model learned finished and model saved, in "model.predict" command, i got error. that error is:
Unknown layer: SpatialPyramidPooling
I run test_spp.py file and it worked. In my model, what should i do?
Thanks

@Kissadada
Copy link

load model by this way
model = load_model('/data/spp-mini_train.hdf5', custom_objects={'SpatialPyramidPooling':SpatialPyramidPooling})

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

No branches or pull requests

2 participants