美文网首页
Detectron-Cascade-RCNN的安装

Detectron-Cascade-RCNN的安装

作者: greathope | 来源:发表于2019-04-17 16:02 被阅读0次
  • 默认配置为ubuntu 16.04, cuda和cudnn都已经安装好了
  1. 安装caffe2
    • 跟着命令一步步敲
  2. clone 官方repo
  3. 安装还是参照Detectron的说明
    • 但注意make还是在Detectron-Cascade-RCNN里做
  4. 如果测试时,出现错误: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')
  5. 安装完成,可以训练模型

相关文章

网友评论

      本文标题:Detectron-Cascade-RCNN的安装

      本文链接:https://www.haomeiwen.com/subject/osunwqtx.html