CTPN - 训练

源码地址: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 上搜索,总会找到解决方法。

 tf 不同版本的问题:https://github.com/CharlesShang/TFFRCNN/issues/25 (作者的代码中也有写到)--- (tf.image_summary should be renamed to tf.summary.image;)

https://github.com/eragonruan/text-detection-ctpn/issues/198  (VGG_imagenet.npy model 不可用)

 

demo中用到的预训练好的模型放在checckpoints 目录中   

posted @ 2018-11-12 15:06  静悟生慧  阅读(2574)  评论(1编辑  收藏  举报