Skip to main content

Image classification models for Gluon

Project description

Large-scale image classification networks

Several large-scale image classification models on MXNet/Gluon, trained on the ImageNet-1k dataset.

Installation

To install, use:

pip install gluoncv2 mxnet>=1.2.1

To enable different hardware supports such as GPUs, check out mxnet variants. For example, you can install with cuda-9.2 supported mxnet:

pip install gluoncv2 mxnet-cu92>=1.2.1

Usage

Example of using the pretrained ResNet-18 model:

from gluoncv2.model_provider import get_model as glcv2_get_model
net = glcv2_get_model("resnet18", pretrained=True)

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

gluoncv2-0.0.4.tar.gz (36.0 kB view hashes)

Uploaded Source

Built Distribution

gluoncv2-0.0.4-py2.py3-none-any.whl (61.8 kB view hashes)

Uploaded Python 2 Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page