GluonCV: a Deep Learning Toolkit for Computer Vision
========================================================
GluonCV provides implementations of state-of-the-art (SOTA) deep learning
algorithms in computer vision.
It is designed for helping engineers, researchers, and students to quickly
prototype products, validate new ideas, and learning computer vision.
GluonCV features:
1. training scripts that reproduces SOTA results reported in latest papers,
2. a large set of pre-trained models,
3. carefully designed APIs and easy to understand implementations,
4. community support.
GluonCV tutorials assume users have basic knowledges about deep learning and
computer vision.
Otherwise, please refer to our introductory deep learning course
`MXNet-the-Streight-Dope `_.
.. note::
This project is at an early stage. Please expect frequent updates.
We welcome feedback and contributions.
Installation
------------------
Install via PyPI
^^^^^^^^^^^^^^^^
The easiest way to install GluonCV is through `pip `_.
.. code-block:: bash
pip install gluoncv
Install from Source
^^^^^^^^^^^^^^^^^^^
Optionally, you can clone the GluonCV project and install it locally
.. code-block:: bash
git clone https://github.com/dmlc/gluon-cv
cd gluon-cv && python setup.py install --user
Install MXNet
^^^^^^^^^^^^^
GluonCV depends on the recent version of MXNet. The easiest way to install MXNet
is through `pip `_. The following
command installs a nightly build CPU version of MXNet.
.. code-block:: bash
pip install --pre mxnet
.. note::
There are other pre-build MXNet binaries that enable GPU support and
accelerate CPU performance, please refer to `this tutorial
`_ for details.
Some training scripts are recommended to run on GPUs, if you don't have a GPU
machine at hands, you may consider to `run on AWS
`_.
A Quick Example
----------------
:ref:`Object Detection Demo `
.. toctree::
:maxdepth: 2
:caption: Tutorials
build/examples_classification/index
build/examples_detection/index
build/examples_segmentation/index
build/examples_datasets/index
.. toctree::
:maxdepth: 1
:caption: Model Zoo
model_zoo/index
.. toctree::
:maxdepth: 1
:caption: API Reference
api/datasets
api/transforms
api/models
api/utils
.. toctree::
:maxdepth: 1
:caption: Community
how_to/support
how_to/contribute