Open RabbearSu opened 5 years ago
请问您当时训练的时候有没有出现损失不收敛的情况?我安装楼主提供的源码调通代码,学习参数未改变的情况下训练一晚上,8个epoch,损失还是不收敛。。望解答。
请问您当时训练的时候有没有出现损失不收敛的情况?我安装楼主提供的源码调通代码,学习参数未改变的情况下训练一晚上,8个epoch,损失还是不收敛。。望解答。
我没有跑楼主的代码,只是参考了他anchor生成的代码。 不能帮到您,不好意思!
https://github.com/AstarLight/Lets_OCR/blob/cce8f8b857e9f8e196bf82cfd0c1ff3656cd445e/detector/ctpn/lib/generate_gt_anchor.py#L30 你要注意第30行,这个range的范围是left_anchor_num,right_anchor_num-1,就把right_anchor_num的值自动忽略了。 改成range(left_anchor_num,right_anchor_num+1)可能可读性更好。
https://github.com/AstarLight/Lets_OCR/blob/cce8f8b857e9f8e196bf82cfd0c1ff3656cd445e/detector/ctpn/lib/generate_gt_anchor.py#L23
这样会减少下一步筛选的量,因为right_anchor_num的值应该是math.floor而不是ceil定位到的值