CTPN学习记录-day1 – Qmshao – Python量化投资

CTPN学习记录-day1 – Qmshao

最近对CTPN的TensorFlow代码进行学习使用

参考了《与CPTN(文字识别网络)作斗争的记录》这篇文章[1],把代码跑了起来

我没有跑文章[1]中提到的进行过改动的另一版CTPN代码,而是将最原始的那版代码在其基础上进行debug(因为某些必要的原因)

这版代码的问题与文章中提到的部分问题是吻合的,甚至问题更少一点(?)

并且通过tensorboard查看两者架构差异的时候发现除了核心的双向LSTM,在其他位置好像使用了不同的操作

并且[1]中运行的代码节点命名的可读性也很差

运行环境为Win10,Anaconda3

修改内容1


nms.pyx 第25行代码

参考文章[1]将nms.pyx第25行代码中的int_t改成intp_t,否则将会报错


bug 1

修改内容2

在ctpn/、ctpn/utils、ctpn/utils/bbox等目录下均创建了__init__.py,因为pylint会瞎报错,并且在跑的时候没有__init__.py会报错(有时候报有时候又没有。。。)

结果

在进行了上述修改后,直接跑main/demo.py就可以得到想要的结果了


demo 1

demo 2

可以看到框选效果还是很不错的

之后将试着实现模型的固化,以及mo转化优化。

© 著作权归作者所有,转载或内容合作请联系作者
https://www.jianshu.com/p/d04cfe0af9c9

「点点赞赏,手留余香」

    还没有人赞赏,快来当第一个赞赏的人吧!
0 条回复 A 作者 M 管理员
    所有的伟大,都源于一个勇敢的开始!
欢迎您,新朋友,感谢参与互动!欢迎您 {{author}},您在本站有{{commentsCount}}条评论