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 中文文档 https://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
原文:https://blog.csdn.net/yql_617540298/article/details/82970478
版权声明:本文为博主原创文章,转载请附上博文链接!
|
|