何恺明 Detectron安装与Mask RCNN介绍 detectron安装+caffe2安装

转载https://blog.csdn.net/Xiongchao99/article/details/79167353/

https://www.cnblogs.com/zealousness/p/8757353.html

 

 

detectron安装+caffe2安装

因为想跑一下facebook最近开源的detectron物体检测平台，所以安装caffe2+detectron

总结：

一定要好好看官方安装教程：https://github.com/facebookresearch/Detectron/blob/master/INSTALL.md

caffe2教程，caffe2与caffe区别：https://www.jianshu.com/p/5c0fd1c9fef9?from=timeline

detectron是基于caffe2框架，使用mask-rcnn模型实现的物体检测平台

github链接：https://github.com/facebookresearch/Detectron

 

detectron demo：

1、使用detectron进行识别：使用tools/infer_simple.py，调用一个已经训练好的模型model_final.pkl，对图片进行物体检测

2、使用detectron进行训练：使用tools/train_net.py

https://github.com/facebookresearch/Detectron/blob/master/GETTING_STARTED.md

 

detectron的依赖：

因为detectron基于caffe2，而且暂时支持GPU计算，所以我们需要安装caffe2，cuda（cudnn）

 

caffe2安装：

上caffe2官网看安装教程：https://caffe2.ai/docs/getting-started.html?platform=windows&configuration=compile

发现可以有conda安装、github源码编译安装等多种方式。