Closed moyemoji closed 4 years ago
@AstarLight 你好，
感谢你的非常详实的博客讲解以及完整的代码。我下载你的项目试着跑了下，成功了让我特别开心。但是训练过程有个问题，就是训练速度很慢，几乎要6s以上才能完成一张图片的训练。
我仔细分析了一下你的源码，发现之所以训练速度慢在于generate_gt_anchor中求取anchor上下边界时以及tag_anchor中计算IOU的时候采取的方式是遍历图像栅格实现。
我用几何计算来实现上述的两种操作，速度有了很明显的提升，大约是0.3-0.6s一张图。
我写的代码可能还有欠考虑的地方，还请@AstarLight 指正！
@AstarLight 你好，
感谢你的非常详实的博客讲解以及完整的代码。我下载你的项目试着跑了下，成功了让我特别开心。但是训练过程有个问题，就是训练速度很慢，几乎要6s以上才能完成一张图片的训练。
我仔细分析了一下你的源码，发现之所以训练速度慢在于generate_gt_anchor中求取anchor上下边界时以及tag_anchor中计算IOU的时候采取的方式是遍历图像栅格实现。
我用几何计算来实现上述的两种操作，速度有了很明显的提升，大约是0.3-0.6s一张图。
我写的代码可能还有欠考虑的地方，还请@AstarLight 指正！