search

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

deltas维度错误 #6

Closed RabbearSu closed 5 years ago

RabbearSu commented 5 years ago

https://github.com/yizt/keras-ctpn/blob/6b962f833bc3abc6e2b19aafa288738130c6b735/ctpn/layers/losses.py#L68

应为 deltas = tf.gather_nd(deltas[..., :-1], positive_indices)，deltas为3维

yizt commented 5 years ago

@RabbearSu 感谢您的反馈！这里只做垂直方向的中心点偏移和高度回归，所以取:-2取到(dy,dh)没有错。

RabbearSu commented 5 years ago

@RabbearSu 感谢您的反馈！这里只做垂直方向的中心点偏移和高度回归，所以取:-2取到(dy,dh)没有错。

好的，我又看了一下，您说的没错，谢谢~