目标检测大菜鸟第一次Faster-RCNN DEMO(笔记)

来源:互联网 发布:淘宝扩容128g u盘判断 编辑:程序博客网 时间:2024/05/17 22:38

说明:纯菜鸟,师兄师姐指导,随记

Faster-RCNN源码下载地址:

Matlab版本:https://github.com/ShaoqingRen/faster_rcnn

Python版本:https://github.com/rbgirshick/py-faster-rcnn


       本次用到的是Python版本,在Linux下运行。  CPU!!

       本文为在师兄blog(http://blog.csdn.net/ture_dream/article/details/52702504)的基础上根据自己运行时出现的一些问题的一次随记。以便自己以后翻阅,同时希望能帮助到遇到同样问题的同学们。


准备工作:

       caffe的配置。

       我用的是Ubuntu16.04 cuda8.0 在Linux上配置的caffe,虽说很简单,配置起来就是一些傻瓜操作,但也遇到了一些棘手的问题。有时间整理caffe的配置,供给以后的师妹看,装装逼。生气

       建议和我一样的Linux菜鸟在操作Linux系统之前先在实验楼学习一点简单的Linux指令(直接百度实验楼,第一个链接就是)

                


下面开始介绍Faster-RCNN+ZF的运行步骤:

      第一步安装cython,python-OpenCV,easydict。

   打开终端(ctrl+Alt+t),依次输入以下命令:

  1. pip install cython  
  2. pip install easydict  
  3. apt-get install python-opencv

     在执行第三条命令的时候,遇到了以下问题:

         

   提示权限不够。只要在命令语句前加上sudo就可以了,如下图:

        

    第二步下载py-faster-rcnn

     在终端内接着输入以下命令:

  1. # Make sure to clone with --recursive  
  2. git clone --recursive https://github.com/rbgirshick/py-faster-rcnn.git

       得到下图:

        

      第三步:进入py-faster-rcnn/lib

        首先找到lib文件

         

        接着按ctrl+L

        

           然后把文件路径复制到终端,用cd进入即可:

       

        (以上步骤是写给和我一样的菜鸟的)

   

        进入py-faster-rcnn/lib之后,我们就可以执行make了。

          

             我在这里出现了错误提示:The CUDA lib64 path could not be located in /usr/lib64

       可以进行以下操作:

           1. gedit 打开 setup.py:

             

           右击setup.py,点击用gedit打开。

        2.找到下图代码所在的位置   

         把lib64改成lib

  

       最后保存即可。 The CUDA lib64 path could not be located in /usr/lib64 这个错误就解决了。

       然后我们继续上面的步骤,来执行make:

                  


     第四步:进入py-faster-rcnn\caffe-fast-rcnn

                   执行 cp Makefile.config.example Makefile.config

               


        接着,配置Makefile.config文件。就是在py-faster-rcnn/caffe-fast-rcnn中找到Makefile.config。如果没有这个文件的话可以复制Makefile.config.example,然后把复制的文件名改成Makefile.config。

       配置的内容

         1.把第二个CPU_ONLY := 1前的#去掉:

                

           2.把 WITH_PYTHON_LAYER := 1前的#去掉:

            

         点击保存,即可。

       然后执行命令:   make -j8 && make pycaffe

     

     这里又出现了问题:

         In file included from src/caffe/solvers/sgd_solver.cpp:5:0:
         ./include/caffe/util/hdf5.hpp:6:18: fatal error: hdf5.h: 没有那个文件或目录
         compilation terminated.

      参考以下步骤:

     



    再执行一次  make -j8 && make pycaffe:

                这就OK了。


     下面就是跑demo了:

                       

       上面显示错误:~/py-faster-rcnn/tools$ install the python-tk package
install: 目标'package' 不是目录。

      

            

 

这里就需要下载安装python-tk了,执行以下指令:

       sudo apt-get install python-tk即可:

       

        

       下载好后,继续执行./demo.py,会有提示:

        WARNING: Logging before InitGoogleLogging() is written to STDERR
F0708 09:04:59.452781  7068 common.cpp:66] Cannot use GPU in CPU-only Caffe: check mode.
*** Check failure stack trace: ***
已放弃 (核心已转储)

       可以执行下图命令:./demo.py --cpu

     

      接着又出现了问题:(图中上半部分为出现的问题,执行 sudo apt-get install python-yaml,可以解决该问题)

     


     好了,再次执行./demo.py --CPU:

    

    最后可以看到:

     

    过段时间桌面就可以蹦出几张图片:

     


附:训练模型从师兄那拷过来的。


阅读全文
2 0
原创粉丝点击