Detectron-Cascade-RCNN的安装
- 默认配置为ubuntu 16.04, cuda和cudnn都已经安装好了
- 安装caffe2
- clone 官方repo
- 安装还是参照Detectron的说明
- 但注意
make
还是在Detectron-Cascade-RCNN
里做
- 如果测试时,出现错误:
Detectron ops lib not found; make sure that your Caffe2 version includes Detectron module
- 在
detectron/utils/env.py
文件的import sys
后追加:
sys.path.insert(0, '/path/to/your/pytorch/build')
- 安装完成,可以训练模型
本文标题:Detectron-Cascade-RCNN的安装
本文链接:https://www.haomeiwen.com/subject/osunwqtx.html
网友评论