yizt / keras-ctpn

keras复现场景文本检测网络CPTN: 《Detecting Text in Natural Image with Connectionist Text Proposal Network》;欢迎试用,关注,并反馈问题...
Apache License 2.0
107 stars 38 forks source link

看源码时有一个疑问 #23

Closed kaixinbaba closed 4 years ago

kaixinbaba commented 4 years ago
def smooth_l1_loss(y_true, y_predict, sigma2=9.0):
    abs_diff = tf.abs(y_true - y_predict, name='abs_diff')
    loss = tf.where(tf.less(abs_diff, 1. / sigma2), 0.5 * sigma2 * tf.pow(abs_diff, 2), abs_diff - 0.5 / sigma2)
    return loss

在源码中有一个关于smooth l1的loss的函数,我网上查询到的smooth l1的定义是 image 当函数sigma2=1的时候和网上定义的是一致的, 但是我看大佬这里用的是9.0的默认值,说实话以我的水平第一眼觉得是不是应该是0.9? 想请问大佬这里sigma2的默认值为什么要设置成9.0,有什么用意或者是经过测试这个数字比较好吗

yizt commented 4 years ago

@kaixinbaba 这个sigma2=9.0是借鉴的其他人的,没有做实验;

kaixinbaba commented 4 years ago

好的 谢谢