GluonCV: 计算机视觉的深度学习工具包¶
GluonCV 提供了计算机视觉领域最先进 (SOTA) 深度学习算法的实现。它旨在帮助工程师、研究人员和学生快速构建产品原型、验证新想法以及学习计算机视觉。
GluonCV 特性
可复现最新论文中报告的 SOTA 结果的训练脚本，
大量的预训练模型，
精心设计的 API 和易于理解的实现，
社区支持。
支持的应用¶
| 应用 | 示例 | 可用模型 |
|---|---|---|
|
图像分类 识别图像中的对象 。 |
 |
50+ 个模型，包括 ResNet, MobileNet, DenseNet, VGG, ... |
| 目标检测 检测多个对象 及其边界框 。 |
|
Faster RCNN, SSD, Yolo-v3 |
| 语义分割 将每个像素关联 图像的 到一个类别标签。 |
|
FCN, PSP, DeepLab v3 |
| 实例分割 将每个像素关联 图像的 到一个实例标签。 |
|
Mask RCNN |
| 姿态估计 检测人体姿态 。 |
|
Simple Pose |
| 视频动作识别 识别视频中的人体动作 。 |
|
TSN, I3D, Non-local, SlowFast |
安装¶
选择您的偏好并运行安装命令。
操作系统： Linux macOS Windows
版本： 稳定版 每夜构建版 源码
稳定发布版。
包含最新功能的每夜构建版。
从源码安装 GluonCV。
后端： 原生 CUDA MKL-DNN CUDA + MKL-DNN
用于 CPU 的内置后端。
在 Nvidia GPU 上运行所需。
加速 Intel CPU 性能。
同时启用 Nvidia GPU 和 Intel CPU 加速。
先决条件
需要 pip >= 9.。
请注意，您可以通过将“pip install gluoncv”替换为“pip install gluoncv[full]”来一并安装所有额外的可选依赖项。
每夜构建版为爱好者提供最新功能。
命令
# for mxnet
pip install --upgrade mxnet
# for pytorch
pip install torch==1.6.0+cpu torchvision==0.7.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
pip install --upgrade gluoncv
# Here we assume CUDA 10.2 is installed. You can change the number
# according to your own CUDA version.
# for mxnet
pip install --upgrade mxnet-cu102
# for pytorch
pip install torch==1.6.0 torchvision==0.7.0
pip install --upgrade gluoncv
pip install --upgrade mxnet-mkl gluoncv
# Here we assume CUDA 10.2 is installed. You can change the number
# according to your own CUDA version.
pip install --upgrade mxnet-cu102mkl gluoncv
# for mxnet
pip install --upgrade mxnet -f https://dist.mxnet.io/python/all
# for pytorch
pip install --pre torch torchvision torchaudio -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
pip install --pre --upgrade gluoncv
# mxnet
pip install --upgrade mxnet-cu102 -f https://dist.mxnet.io/python/all
# pytorch
pip install --pre torch torchvision torchaudio -f https://download.pytorch.org/whl/nightly/cu102/torch_nightly.html
pip install --pre --upgrade gluoncv
pip install --pre --upgrade gluoncv
pip install --upgrade mxnet-mkl -f https://dist.mxnet.io/python/all
pip install --pre --upgrade gluoncv
pip install --upgrade mxnet-cu102mkl -f https://dist.mxnet.io/python/all
# mxnet
pip install --upgrade mxnet -f https://dist.mxnet.io/python/all
# pytorch
pip install --pre torch torchvision torchaudio -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
git clone https://github.com/dmlc/gluon-cv
cd gluon-cv && python setup.py install --user
# mxnet
pip install --upgrade mxnet-cu100 -f https://dist.mxnet.io/python/all
# pytorch
pip install --pre torch torchvision torchaudio -f https://download.pytorch.org/whl/nightly/cu102/torch_nightly.html
git clone https://github.com/dmlc/gluon-cv
cd gluon-cv && python setup.py install --user
pip install --upgrade mxnet-mkl -f https://dist.mxnet.io/python/all
git clone https://github.com/dmlc/gluon-cv
cd gluon-cv && python setup.py install --user
pip install --upgrade mxnet-cu102mkl -f https://dist.mxnet.io/python/all
git clone https://github.com/dmlc/gluon-cv
cd gluon-cv && python setup.py install --user
查看 安装 获取更多安装说明和选项。
深度学习或计算机视觉新手？¶
有关深度学习或计算机视觉的背景知识，请参考开源书籍 Dive into Deep Learning。
