【背景】Facebook 开源了物体检测研究平台 Detectron。Detectron 是 Facebook AI 研究的软件系统,实现了最先进的物体检测算法,包括 Mask R-CNN。Detectron 用Python 编写实现,并由深度学习框架 Caffe2 驱动。
Caffe2安装:(官网https://caffe2.ai/)有conda安装、github源码编译安装等多种方式。
1. conda安装
conda安装的caffe2无detectron模块,需要最新的caffe2支持,因此不能用conda安装。
2. 使用源码安装
(1)pytorch源码
前不久Caffe2 开源代码并入 PyTorch,caffe2官网提供的源码编译安装自pytorch
进行到sudo make install,错误如下:
解决:建立build之前
1)将CMakeLists.txt中local protobuf改为OFF
2)终端cmake -DCAFFE2_LINK_LOCAL_PROTOBUF=OFF
(2)caffe2源码
caffe2目录下make时,错误如下:
解决:
1)将CMakeLists.txt中MPI改为OFF
2)终端cmake .. -DUSE_MPI=OFF
最后,测试,输出Success成功。
其他问题:打开python,输入import caffe后出现ImportError: No module named caffe,此时添加环境变量,在文尾输入,保存
领取专属 10元无门槛券
私享最新 技术干货