Skip to main content

Gluon CV Toolkit

Project description

GluonCV provides implementations of the state-of-the-art (SOTA) deep learning models in computer vision.

It is designed for engineers, researchers, and students to fast prototype products and research ideas based on these models.


To install, use:

pip install gluoncv mxnet>=1.6.0 --upgrade
# for installing gluoncv with all dependencies
pip install gluoncv[full] mxnet>=1.6.0 --upgrade

To enable different hardware supports such as GPUs, check out mxnet variants.

For example, you can install cuda-11.0 supported mxnet alongside gluoncv:

pip install gluoncv mxnet-cu110>=1.6.0 --upgrade

Project details

Release history Release notifications | RSS feed