基于tensorflow的CTPN实现

申明,本文是跟在https://github.com/eragonruan/text-detection-ctpn这个作者后面做的

最近在复现文字检测,因为caffe环境一直没搭好,所以想先在已有的条件下做一个小的实验,是基于tensorflow的

1.参照前辈给出的实现代码,安装依赖的库文件,参照requirements.txt

easydict==1.7

tensorflow_gpu==1.3.0

scipy==0.18.1

numpy==1.11.1

opencv_python==3.4.0.12

Cython==0.27.3

Pillow==5.0.0

PyYAML==3.12

注意:这里踩到的坑是,明明执行了pip install tensorflow但是在执行.py脚本时报错:No module named tensorflow ,这个可能的解决办法是先卸载protobu,再安装一次

2.接下来,就是cd ~/text-detection-ctpn/ctpn目录下

3.输入 python ./ctpn/demo.py如果执行没有报错,说明成功,可以在result里面看到标注文字的图片

我不放心这个结果,自己在网上找了些照片放到data/demo文件夹下面,再跑一次2-3发现新加入的图片也能检测出来,说明确实有效

后面我会继续这个实验,自己找数据集,训练模型,看看效果再来更新

你可能感兴趣的:(基于tensorflow的CTPN实现)