CTPN - 训练

Posted 2021-01-19 allen-rg

源码地址：https://github.com/eragonruan/text-detection-ctpn

该地址提供了 CTPN 的 tf 版本的实现，代码文档写得很详细，issue 里面也帮助解决了不少问题。

下面简单记录在复现训练的时候遇到的一些问题：

1、首先，必要的环境配置

2、可以尝试用已有的训练好的模型，运行 demo.py 测试一些图片

3、训练数据准备，按照作者的要求，将他已经准备好的数据（一定的格式）拷贝到data/ 目录下面，然后将对应的名字命名正确，能找到对应的文件

4、训练时， _image_summary  的位置报错，是作者为了提醒我们，用不同的TF版本，最终所用的函数名是有区别的，采用合适的版本即可

5、最后 VGG_imagenet.npy  文件时空 的，需要下载预训练好的model，放在指定的目录下面，来完成训练

 

总之，需要细心，报错之后按照错误，在git上或者Google 上搜索，总会找到解决方法。