![](https://img.haomeiwen.com/i8207483/238a55044d489f72.png)
最近一直在搞目标检测,用了很多框架,今天开始说一说用起来比较顺手的框架吧。今天主角是 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 来跑你的模型。
网友评论