smooth_l1_loss(y_true, y_pred)
diff = K.abs(y_true - y_pred)
less_than_one = K.cast(K.less(diff, 1.0), "float32")
loss = (less_than_one * 0.5 * diff**2) + (1 - less_than_one) * (diff - 0.5)
return loss
参考链接： Keras 中文文档 keras.io/zh/backend/
一、keras.backend.less()
keras.backend.less(x, y)
参数
输入两个张量
返回值
逐个元素比对（）的真值。
二、keras.backend.cast()
keras.backend.cast(x, dtype)
参数
（1）x：张量；
（2）dtype：字符串，'float16'，'float32'，'float64'；
返回值
将张量转换到不同的dtype并返回。
---------------------
作者：蹦跶的小羊羔
来源：CSDN
原文：blog.csdn.net/yql_6175402…
版权声明：本文为博主原创文章，转载请附上博文链接！ |
|
