Stuck on an issue?

Lightrun Answers was designed to reduce the constant googling that comes with debugging 3rd party libraries. It collects links to all the places you might be looking at while hunting down a tough bug.

And, if you’re still stuck at the end, we’re happy to hop on a call to see how we can help out.

"ValueError: Unknown layer: InstanceNormalization" while predicting (isensee2017)

See original GitHub issue

Hello, I have followed the readme to train the brats data using the method of isensee2017. The training phrase worked well, but when performing python predict.py, an error occurred

ValueError: Unknown layer: InstanceNormalization

at https://github.com/ellisdg/3DUnetCNN/blob/master/unet3d/prediction.py#L120

I’ve searched in Google but found no results, could you please give me some ideas?

Issue Analytics

State:
Created 6 years ago
Comments:5 (2 by maintainers)

Top GitHub Comments

4reactions

ellisdgcommented, Jan 10, 2018

This has to do with the load_model function not understanding the Instance Normalization layer. I’ll try to fix this soon, but to fix it now this you can add from keras-contrib import InstanceNormalization to the training.py file and then in the load_model function add 'InstanceNormalization': InstanceNormalization to the custom_objects dictionary. When loading the model this will tell Keras what exactly ‘InstanceNormalization’ is referring to.

0reactions

b-zcommented, Jan 30, 2018

Thanks a lot!