深度学习基础篇【1】
从0开始搭建
Detectron2
进行简单的图像识别
detectron2是facebook AI research(FAIR)重构detectron的深度学习框架,是2020年最火的深度学习框架(另外还有mmdetection,simpledet,目标检测三大开源神器:Detectron2/mmDetection/SimpleDet。
detectron2从基于caffe重构为完全基于pytorch,集成了先进的目标检测和语义分割算法。
本文搭建Detectron2,并进行简单的demo实验。
- 语言环境:Python3.8
- 编译器:jupyter notebook
- 深度学习环境:PyTorch 1.8
- CUDA版本:11.1.1
- 深度学习平台:极链AI云
- 显卡(GPU):NVIDIA RTX 2080 Ti