cafffe+faster rcnn 基于docker的安装

安装教程参照https://blog.csdn.net/flygeda/article/details/78638824

dzy@smiles-SYS-4028GR-TR2-1-EC028:~$ docker run --runtime=nvidia -it --ipc=host -v /data1/lyh:/hom                                   e/lyh --name lyh2_caffe ba28bcb1294c   bash

root@4f751951981d:/workspace# pip install opencv-python
Collecting opencv-python
Installing collected packages: numpy, opencv-python
  Found existing installation: numpy 1.11.0
    Uninstalling numpy-1.11.0:
      Successfully uninstalled numpy-1.11.0
Successfully installed numpy-1.16.6 opencv-python-4.2.0.32

root@4f751951981d:/workspace# pip install Cython
Requirement already satisfied: Cython in /usr/local/lib/python2.7/dist-packages

root@4f751951981d:/workspace# pip install easydict
Collecting easydict
Successfully installed easydict-1.9

root@4f751951981d:/home/lyh/Re_2020/Caffe_deep# git clone --recursive https://github.com/rbgirshic                                   k/py-faster-rcnn.git

root@4f751951981d:/home/lyh/Re_2020/Caffe_deep# cd py-faster-rcnn/
root@4f751951981d:/home/lyh/Re_2020/Caffe_deep/py-faster-rcnn# cd lib/
root@4f751951981d:/home/lyh/Re_2020/Caffe_deep/py-faster-rcnn/lib# make

root@4f751951981d:/home/lyh/Re_2020/Caffe_deep/py-faster-rcnn/lib# cd ..
root@4f751951981d:/home/lyh/Re_2020/Caffe_deep/py-faster-rcnn# cd caffe-fast-rcnn/

root@4f751951981d:/home/lyh/Re_2020/Caffe_deep/py-faster-rcnn/caffe-fast-rcnn# cp Makefile.config.                                   example Makefile.config
root@4f751951981d:/home/lyh/Re_2020/Caffe_deep/py-faster-rcnn/caffe-fast-rcnn# pkg-config --modver                                   sion opencv
2.4.9.1

root@4f751951981d:/home/lyh/Re_2020/Caffe_deep# chmod -R 777 py-faster-rcnn/

root@4f751951981d:/home/lyh/Re_2020/Caffe_deep# cp /usr/local/cuda/include/host_config.h host_conf                                   ig.h

root@4f751951981d:/home/lyh/Re_2020/Caffe_deep# chmod 777 host_config.h

root@4f751951981d:/home/lyh/Re_2020/Caffe_deep# cp host_config.h /usr/local/cuda/include/host_conf                                   ig.h

root@4f751951981d:/home/lyh/Re_2020/Caffe_deep# cp /usr/local/cuda/include/host_config.h host_conf                                   ig.h
root@4f751951981d:/home/lyh/Re_2020/Caffe_deep# cd py-faster-rcnn/caffe-fast-rcnn/
#make之前移动caffe源码部分文件到caffe-faster-rcnn
root@4f751951981d:/home/lyh/Re_2020/Caffe_deep/py-faster-rcnn/caffe-fast-rcnn# make -j8 && make py                                   caffe

问题一

oot@4f751951981d:/home/lyh/Re_2020/Caffe_deep/py-faster-rcnn# ./tools/demo.py
Traceback (most recent call last):
  File "./tools/demo.py", line 21, in <module>
    import matplotlib.pyplot as plt
  File "/usr/local/lib/python2.7/dist-packages/matplotlib/pyplot.py", line 115, in <module>
    _backend_mod, new_figure_manager, draw_if_interactive, _show = pylab_setup()
  File "/usr/local/lib/python2.7/dist-packages/matplotlib/backends/__init__.py", line 63, in pylab_setup
    [backend_name], 0)
  File "/usr/local/lib/python2.7/dist-packages/matplotlib/backends/backend_tkagg.py", line 4, in <module>
    from . import tkagg  # Paint image to Tk photo blitter extension.
  File "/usr/local/lib/python2.7/dist-packages/matplotlib/backends/tkagg.py", line 5, in <module>
    from six.moves import tkinter as Tk
  File "/usr/local/lib/python2.7/dist-packages/six.py", line 203, in load_module
    mod = mod._resolve()
  File "/usr/local/lib/python2.7/dist-packages/six.py", line 115, in _resolve
    return _import_module(self.mod)
  File "/usr/local/lib/python2.7/dist-packages/six.py", line 82, in _import_module
    __import__(name)
  File "/usr/lib/python2.7/lib-tk/Tkinter.py", line 42, in <module>
    raise ImportError, str(msg) + ', please install the python-tk package'
ImportError: No module named _tkinter, please install the python-tk package

python-tk 库安装失败，在此处找到答案。如下图所示：

问题二

编译时报错

python/caffe/_caffe.cpp:11:31: fatal error: numpy/arrayobject.h: No such file or directory
compilation terminated.
Makefile:489: recipe for target 'python/caffe/_caffe.so' failed
make: *** [python/caffe/_caffe.so] Error 1
root@4f751951981d:/home/lyh/Re_2020/Caffe_deep/py-faster-rcnn/caffe-fast-rcnn# python
Python 2.7.12 (default, Dec  4 2017, 14:50:18)
[GCC 5.4.0 20160609] on linux2
Type "help", "copyright", "credits" or "license" for more information.
>>> import numpy as np
>>> np.get_include()
'/usr/local/lib/python2.7/dist-packages/numpy/core/include'

参考博客https://blog.csdn.net/wuzuyu365/article/details/52430657
make.config里修改为/usr/local/lib/python2.7/dist-packages/numpy/core/include

问题三

ImportError: cannot import name _validate_lengths

root@4f751951981d:/home/lyh/Re_2020/Caffe_deep/py-faster-rcnn# ./tools/demo.py
Traceback (most recent call last):
  File "./tools/demo.py", line 18, in <module>
    from fast_rcnn.test import im_detect
  File "/home/lyh/Re_2020/Caffe_deep/py-faster-rcnn/tools/../lib/fast_rcnn/test.py", line 16, in <                                   module>
    import caffe
  File "/home/lyh/Re_2020/Caffe_deep/py-faster-rcnn/tools/../caffe-fast-rcnn/python/caffe/__init__                                   .py", line 1, in <module>
    from .pycaffe import Net, SGDSolver, NesterovSolver, AdaGradSolver, RMSPropSolver, AdaDeltaSol                                   ver, AdamSolver
  File "/home/lyh/Re_2020/Caffe_deep/py-faster-rcnn/tools/../caffe-fast-rcnn/python/caffe/pycaffe.                                   py", line 15, in <module>
    import caffe.io
  File "/home/lyh/Re_2020/Caffe_deep/py-faster-rcnn/tools/../caffe-fast-rcnn/python/caffe/io.py",                                    line 2, in <module>
    import skimage.io
  File "/usr/local/lib/python2.7/dist-packages/skimage/__init__.py", line 158, in <module>
    from .util.dtype import *
  File "/usr/local/lib/python2.7/dist-packages/skimage/util/__init__.py", line 7, in <module>
    from .arraycrop import crop
  File "/usr/local/lib/python2.7/dist-packages/skimage/util/arraycrop.py", line 8, in <module>
    from numpy.lib.arraypad import _validate_lengths
ImportError: cannot import name _validate_lengths

root@4f751951981d:/home/lyh/Re_2020/Caffe_deep/py-faster-rcnn# python -c "import numpy; print nump                                   y.__version__"
1.16.6

root@4f751951981d:/home/lyh/Re_2020/Caffe_deep/py-faster-rcnn# pip install -U scikit-image

在此处找到答案。

# The following worked fine for me
pip install -U scikit-image

Installing collected packages: scikit-image
Found existing installation: scikit-image 0.14.1
Uninstalling scikit-image-0.14.1:
Successfully uninstalled scikit-image-0.14.1
Successfully installed scikit-image-0.15.0

python -c "import skimage"
# No errors, as expected

附录

https://stackoverflow.com/questions是一个很好的问题解决网站。