Skip to main content

Image classification models for PyTorch

Project description

Large-scale image classification networks

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

Installation

To install, use:

pip install pytorchcv torch>=0.4.1

To enable/disable different hardware supports such as GPUs, check out PyTorch installation instructions.

Usage

Example of using the pretrained ResNet-18 model:

from pytorchcv.model_provider import get_model as ptcv_get_model
net = ptcv_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

pytorchcv-0.0.5.tar.gz (33.9 kB view hashes)

Uploaded Source

Built Distribution

pytorchcv-0.0.5-py2.py3-none-any.whl (61.9 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