ctpn训练自己的数据,

  ctpn训练自己的数据,

  本文介绍了运行ctpn的步骤和几个pit。

  简介:

  CPN用于字符检测。也就是说,找到文字在图像中的位置,显示边框。

  执行代码链接:https://github.com/akshayubhat/ctpn 8

  ctpn原文链接:https://github.com/tianzhi 0549/ctpn基于cuda7.0,但是这个cuda版本已经过时了。我的cuda版本是cuda8.0,所以直接找到了第一个链接,没有使用原作者的代码。

  问题一:cudnn版本问题

  原因:cudnn版本冲突,因为caffe的cudnn*。cpp/cudnn*。cu需要加速,要经常更新。(这个问题和跑得更快的原因一样-rcnn))。

  解决办法:

  1)/py-faster-rcnn/caffe-fast-rcnn/include/caffe/util/Cu dnn . HPP在caffe的最新版本上,即对应的Cu dnn上,替换为cudnn的实现。

  2)/py-fast-rcnn/caffe-fast-rcnn/src/caffe/以所有cudnn开头的层,如cudnn_lrn_layer.cu、cudnn

  在caffe的最新版本中,所有文件都将被替换为同名的相应文件。

  3)所有以cudnn开头的文件。/include/caffe/layers,比如cud nn _ lay.hpp/p _ conv _ layer . HPP,cudnn _ lcn _

  Q2:如果你浏览不同文件夹中的py文件,你不能使用导入py文件,因为找不到它。

  解决办法:

  方法1:

  修改~/中的环境变量。bashrc,然后在source ~/.bashrc中。

  方法二:

  的进口。pth文件

  如果要将路径文件(如mypkpath.pth)添加到站点包,则必须添加要在后面添加的模块文件的目录名。pth。1 windows c:\ python 27 \ site-packages #学生将pth文件直接放在c:\python27#的目录下(或其他Python安装目录下)进行测试# 2015年9月232Linux)。

  /usr/local/lib/python 2.7/dist-packages(notebyshanql,我把它放在这个目录下才有效。要添加的目录必须存在,否则不会成功。)

  如果不行,就放在/usr/lib/python2.7里

  2 Linux(red hat)/usr/lib/python 2.7/站点包

  基坑完工了。开始执行代码。

  请参考。

  执行步骤

  1 )1) ) )。

  git clone-递归https://github.com/fresh前置任务/CTPN.git

  请注意这里。您应该在终端上运行这段代码,而不是下载zip文件并克隆源文件。-因为名为- recursive的参数引用了其他caffe的依赖关系。

  2)替换cudnn*。cpp和cudnn*。cu文件和编译caffe

  全部制作-j32

  生成py咖啡

  3)在项目根目录下编译整个项目,生成cpu_nms.c文件。

  cd /home/xuy/CTPN

  美容化妆品

  4)将项目的caffemodel下载到指定文件夹。

  wget 3358 text det.com/downloads/ctpn _受训_模型.咖啡模型-p模型/

  5)执行代码并输出结果。

  python tools/demo.py

郑重声明:本文由网友发布,不代表盛行IT的观点,版权归原作者所有,仅为传播更多信息之目的,如有侵权请联系,我们将第一时间修改或删除,多谢。

留言与评论(共有 条评论)
   
验证码: