keras自定义损失函数并且模型加载的写法介绍


Posted in Python onJune 15, 2020

keras自定义函数时候,正常在模型里自己写好自定义的函数,然后在模型编译的那行代码里写上接口即可。如下所示,focal_loss和fbeta_score是我们自己定义的两个函数,在model.compile加入它们,metrics里‘accuracy'是keras自带的度量函数。

def focal_loss():
 ...
 return xx
def fbeta_score():
 ...
 return yy
model.compile(optimizer=Adam(lr=0.0001), loss=[focal_loss],metrics=['accuracy',fbeta_score] )

训练好之后,模型加载也需要再额外加一行,通过load_model里的custom_objects将我们定义的两个函数以字典的形式加入就能正常加载模型啦。

weight_path = './weights.h5'
model = load_model(weight_path,custom_objects={'focal_loss': focal_loss,'fbeta_score':fbeta_score})

补充知识:keras如何使用自定义的loss及评价函数进行训练及预测

1.有时候训练模型,现有的损失及评估函数并不足以科学的训练评估模型,这时候就需要自定义一些损失评估函数,比如focal loss损失函数及dice评价函数 for unet的训练。

2.在训练建模中导入自定义loss及评估函数。

#模型编译时加入自定义loss及评估函数
model.compile(optimizer = Adam(lr=1e-4), loss=[binary_focal_loss()],
    metrics=['accuracy',dice_coef])

#自定义loss及评估函数
def binary_focal_loss(gamma=2, alpha=0.25):
 """
 Binary form of focal loss.
 适用于二分类问题的focal loss
 focal_loss(p_t) = -alpha_t * (1 - p_t)**gamma * log(p_t)
  where p = sigmoid(x), p_t = p or 1 - p depending on if the label is 1 or 0, respectively.
 References:
  https://arxiv.org/pdf/1708.02002.pdf
 Usage:
  model.compile(loss=[binary_focal_loss(alpha=.25, gamma=2)], metrics=["accuracy"], optimizer=adam)
 """
 alpha = tf.constant(alpha, dtype=tf.float32)
 gamma = tf.constant(gamma, dtype=tf.float32)

 def binary_focal_loss_fixed(y_true, y_pred):
  """
  y_true shape need be (None,1)
  y_pred need be compute after sigmoid
  """
  y_true = tf.cast(y_true, tf.float32)
  alpha_t = y_true * alpha + (K.ones_like(y_true) - y_true) * (1 - alpha)

  p_t = y_true * y_pred + (K.ones_like(y_true) - y_true) * (K.ones_like(y_true) - y_pred) + K.epsilon()
  focal_loss = - alpha_t * K.pow((K.ones_like(y_true) - p_t), gamma) * K.log(p_t)
  return K.mean(focal_loss)

 return binary_focal_loss_fixed

#'''
#smooth 参数防止分母为0
def dice_coef(y_true, y_pred, smooth=1):
 intersection = K.sum(y_true * y_pred, axis=[1,2,3])
 union = K.sum(y_true, axis=[1,2,3]) + K.sum(y_pred, axis=[1,2,3])
 return K.mean( (2. * intersection + smooth) / (union + smooth), axis=0)

注意在模型保存时,记录的loss函数名称:你猜是哪个

a:binary_focal_loss()

b:binary_focal_loss_fixed

3.模型预测时,也要加载自定义loss及评估函数,不然会报错。

该告诉上面的答案了,保存在模型中loss的名称为:binary_focal_loss_fixed,在模型预测时,定义custom_objects字典,key一定要与保存在模型中的名称一致,不然会找不到loss function。所以自定义函数时,尽量避免使用我这种函数嵌套的方式,免得带来一些意想不到的烦恼。

model = load_model('./unet_' + label + '_20.h5',custom_objects={'binary_focal_loss_fixed': binary_focal_loss(),'dice_coef': dice_coef})

以上这篇keras自定义损失函数并且模型加载的写法介绍就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python 字符串操作实现代码(截取/替换/查找/分割)
Jun 08 Python
Python多进程通信Queue、Pipe、Value、Array实例
Nov 21 Python
Django框架中render_to_response()函数的使用方法
Jul 16 Python
Django接受前端数据的几种方法总结
Nov 04 Python
Django使用Celery异步任务队列的使用
Mar 13 Python
Python爬虫包BeautifulSoup异常处理(二)
Jun 17 Python
python @classmethod 的使用场合详解
Aug 23 Python
Pytorch对Himmelblau函数的优化详解
Feb 29 Python
pycharm部署、配置anaconda环境的教程
Mar 24 Python
windows10 pycharm下安装pyltp库和加载模型实现语义角色标注的示例代码
May 07 Python
python + selenium 刷B站播放量的实例代码
Jun 12 Python
python 根据列表批量下载网易云音乐的免费音乐
Dec 03 Python
python语言是免费还是收费的?
Jun 15 #Python
DataFrame.groupby()所见的各种用法详解
Jun 14 #Python
详解pandas.DataFrame.plot() 画图函数
Jun 14 #Python
Pandas把dataframe或series转换成list的方法
Jun 14 #Python
详解pandas获取Dataframe元素值的几种方法
Jun 14 #Python
Pandas对DataFrame单列/多列进行运算(map, apply, transform, agg)
Jun 14 #Python
Python脚本破解压缩文件口令实例教程(zipfile)
Jun 14 #Python
You might like
那些年我们错过的魔术方法(Magic Methods)
2014/01/14 PHP
PHP小教程之实现双向链表
2014/06/12 PHP
四种php中webservice实现的简单架构方法及实例
2015/02/03 PHP
基于jquery的下拉框改变动态添加和删除表格实现代码
2020/09/12 Javascript
各情景下元素宽高的获取实现代码
2011/09/13 Javascript
从js向Action传中文参数出现乱码问题的解决方法
2013/12/29 Javascript
js代码实现无缝滚动(文字和图片)
2015/08/20 Javascript
js实现兼容性好的微软官网导航下拉菜单效果
2015/09/07 Javascript
一起学写js Calender日历控件
2016/04/14 Javascript
jquery获取所有选中的checkbox实现代码
2016/05/26 Javascript
浅谈JS之iframe中的窗口
2016/09/13 Javascript
laravel5.4+vue+element简单搭建的示例代码
2017/08/29 Javascript
js实现动态改变radio状态的方法
2018/02/28 Javascript
Bootstrap Fileinput 4.4.7文件上传实例详解
2018/07/25 Javascript
15个顶级开源JavaScript框架和库
2018/10/10 Javascript
使用js在layui中实现上传图片压缩
2019/06/18 Javascript
vue 对axios get pust put delete封装的实例代码
2020/01/05 Javascript
如何通过JS实现日历简单算法
2020/10/14 Javascript
[01:29:46]DOTA2上海特级锦标赛C组资格赛#1 OG VS LGD第二局
2016/02/27 DOTA
Python实现二叉树结构与进行二叉树遍历的方法详解
2016/05/24 Python
Python实现的计算器功能示例
2018/04/26 Python
Python3删除排序数组中重复项的方法分析
2019/01/31 Python
Tensorflow加载Vgg预训练模型操作
2020/05/26 Python
Python xlwt模块使用代码实例
2020/06/10 Python
python 使用递归的方式实现语义图片分割功能
2020/07/16 Python
webView加载html图片遇到的问题解决
2019/10/08 HTML / CSS
final, finally, finalize的区别
2012/03/01 面试题
财务会计实习报告体会
2013/12/20 职场文书
《乌塔》教学反思
2014/02/17 职场文书
普通大学毕业生自荐信范文
2014/02/23 职场文书
银行优秀员工事迹材料
2014/05/29 职场文书
我的中国梦演讲稿300字
2014/08/19 职场文书
重阳节活动总结
2014/08/27 职场文书
中学教师个人总结
2015/02/10 职场文书
2015年新农村建设指导员工作总结
2015/07/24 职场文书
先进工作者主要事迹材料
2015/11/03 职场文书