Detectron2

作者: zidea | 来源:发表于2020-08-07 20:16 被阅读0次
Detectron2

最近一直在搞目标检测,用了很多框架,今天开始说一说用起来比较顺手的框架吧。今天主角是 Detectron2,Detectron2 算是一个比较成熟的框架,框架设计除了在模型身上花了一些心事,也为框架的易用性花了一些心思。
我们先从两个方面来说感受,首先是不用我们自己造轮子,为当写流行数据集格式提供了接口。方面我们快速接入数据集到模型模型中。同时也提供接口,或者只要满足其对数据集格式的要求就行,更自由。
现在主流框架从仅追求模型准确率走出来,考虑如何框架快速推广到我们开发者面前。我们主要从几个方面来学习这个框架 Detetron2。

最好的方式就是看官方文档 detectron2
先从安装开始说起

  • 需要 linux 和 macOS 操作系统,window 没有试过
  • python 版本需要 3.6
  • pytorch 需要 1.4 以上版本,最近个人也学学习 pytorch
  • 需要安装 python 版本 opencv
  • CC=clang CXX=clang++ python -m pip install 'git+https://github.com/facebookresearch/detectron2.git' 通过源码来安装 Detectron2
    有条件 Colab Notebook 跑你的程序,在 Colab 上提供 CPU、GPU 还有 TPU,可以使用 Colab 对 GPU 来跑你的模型。

相关文章

网友评论

    本文标题:Detectron2

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