load_model and Lamda layer in Keras

Question

How to load model that have lambda layer?

Here is the code to reproduce behaviour:

MEAN_LANDMARKS = np.load(\'data/mean_shape_68.npy\')

def add_mean_lan         


        

Answer 1

You need to pass custom_objects argument to load_model function:

model = load_model('model_file_name.h5', custom_objects={'MEAN_LANDMARKS': MEAN_LANDMARKS})

Look for more info in Keras docs: Handling custom layers (or other custom objects) in saved models .