一、编译caffe
cd进入hed-master文件夹目里下
(1)
cp Makefile.config.example Makefile.config
(2)
make all
出现错误a
解决:
打开Makefile.config文件
将
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib
修改为:
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu /usr/lib/x86_64-linux-gnu/hdf5/serial
出现错误b
解决:
打开Makefile文件
找到
LIBRARIES += glog gflags protobuf leveldb snappy \
lmdb boost_system boost_filesystem hdf5_hl hdf5 m \
opencv_core opencv_highgui opencv_imgproc
修改为:
LIBRARIES += glog gflags protobuf leveldb snappy \
lmdb boost_system boost_filesystem hdf5_hl hdf5 m \
opencv_core opencv_highgui opencv_imgproc opencv_imgcodecs opencv_videoio
出现错误c
解决:
打开Makefile.config
删除这两行
-gencode arch=compute_20,code=sm_20 \
-gencode arch=compute_20,code=sm_21 \
(3)make test
(4)make runtest
(5)make pycaffe
二、运行
下载所需的数据集和模型,在https://github.com/s9xie/hed中的training HED
在examples--hed中新建一个data文件夹,将解压后的数据集放入data文件夹中,模型就放在examples---hed中。然后打开examples---hed中的train_val.prototxt,
里面有两个root_folder,将这个路径改成你的路径。
三、执行solve.py
在hed-master/examples/hed打开终端
执行
python solve.py
然后出现问题
这是python2和python3中print的错误,所以建立一个Python2的虚拟环境
conda create -n py27 python=2.7
创建虚拟环境可以参考https://pythonguidecn.readthedocs.io/zh/latest/dev/virtualenvs.html
然后再次运行,看缺什么module就pip install
pip install scikit-image
pip install protobuf
然后再次运行solve.py