Keras load model 导入自定义函数

Keras 是一个强大的深度学习框架，我习惯用 ModelCheckpoint 的方式将模型保存为 hdf5 格式，然后再用 load_model 函数加载模型。今天跑实验的时候却报了一个 error，如下：

ValueError: Unknown loss function: dice_coef_loss

原来 load_model 加载的是默认的 loss function, 而 dice_coef_loss 是我自己定义的 loss function，那么怎么调用自定义的函数呢？答案是加一个 custom_objects 参数：

model=load_model('test.hdf5',custom_objects{'dice_coef_loss':dice_coef_loss})

这样问题就解决啦！