Skip to main content

image and video datasets and models for torch deep learning

Project description

torchvision

https://travis-ci.org/pytorch/vision.svg?branch=master https://codecov.io/gh/pytorch/vision/branch/master/graph/badge.svg https://pepy.tech/badge/torchvision https://img.shields.io/badge/dynamic/json.svg?label=docs&url=https%3A%2F%2Fpypi.org%2Fpypi%2Ftorchvision%2Fjson&query=%24.info.version&colorB=brightgreen&prefix=v

The torchvision package consists of popular datasets, model architectures, and common image transformations for computer vision.

Installation

We recommend Anaconda as Python package management system. Please refer to pytorch.org for the detail of PyTorch (torch) installation. The following is the corresponding torchvision versions and supported Python versions.

torch

torchvision

python

master / nightly

master / nightly

>=3.6

1.5.0

0.6.0

>=3.5

1.4.0

0.5.0

==2.7, >=3.5, <=3.8

1.3.1

0.4.2

==2.7, >=3.5, <=3.7

1.3.0

0.4.1

==2.7, >=3.5, <=3.7

1.2.0

0.4.0

==2.7, >=3.5, <=3.7

1.1.0

0.3.0

==2.7, >=3.5, <=3.7

<=1.0.1

0.2.2

==2.7, >=3.5, <=3.7

Anaconda:

conda install torchvision -c pytorch

pip:

pip install torchvision

From source:

python setup.py install
# or, for OSX
# MACOSX_DEPLOYMENT_TARGET=10.9 CC=clang CXX=clang++ python setup.py install

By default, GPU support is built if CUDA is found and torch.cuda.is_available() is true. It’s possible to force building GPU support by setting FORCE_CUDA=1 environment variable, which is useful when building a docker image.

Image Backend

Torchvision currently supports the following image backends:

  • Pillow (default)

  • Pillow-SIMD - a much faster drop-in replacement for Pillow with SIMD. If installed will be used as the default.

  • accimage - if installed can be activated by calling torchvision.set_image_backend('accimage')

C++ API

TorchVision also offers a C++ API that contains C++ equivalent of python models.

Installation From source:

mkdir build
cd build
# Add -DWITH_CUDA=on support for the CUDA if needed
cmake ..
make
make install

Once installed, the library can be accessed in cmake (after properly configuring CMAKE_PREFIX_PATH) via the TorchVision::TorchVision target:

find_package(TorchVision REQUIRED)
target_link_libraries(my-target PUBLIC TorchVision::TorchVision)

The TorchVision package will also automatically look for the Torch package and add it as a dependency to my-target, so make sure that it is also available to cmake via the CMAKE_PREFIX_PATH.

For an example setup, take a look at examples/cpp/hello_world.

Documentation

You can find the API documentation on the pytorch website: http://pytorch.org/docs/master/torchvision/

Contributing

We appreciate all contributions. If you are planning to contribute back bug-fixes, please do so without any further discussion. If you plan to contribute new features, utility functions or extensions, please first open an issue and discuss the feature with us.

Disclaimer on Datasets

This is a utility library that downloads and prepares public datasets. We do not host or distribute these datasets, vouch for their quality or fairness, or claim that you have license to use the dataset. It is your responsibility to determine whether you have permission to use the dataset under the dataset’s license.

If you’re a dataset owner and wish to update any part of it (description, citation, etc.), or do not want your dataset to be included in this library, please get in touch through a GitHub issue. Thanks for your contribution to the ML community!

Project details


Release history Release notifications | RSS feed

Download files

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

Source Distributions

No source distribution files available for this release.See tutorial on generating distribution archives.

Built Distributions

torchvision-0.7.0-cp38-cp38-manylinux1_x86_64.whl (5.9 MB view details)

Uploaded CPython 3.8

torchvision-0.7.0-cp38-cp38-macosx_10_9_x86_64.whl (387.7 kB view details)

Uploaded CPython 3.8macOS 10.9+ x86-64

torchvision-0.7.0-cp37-cp37m-manylinux1_x86_64.whl (5.9 MB view details)

Uploaded CPython 3.7m

torchvision-0.7.0-cp37-cp37m-macosx_10_9_x86_64.whl (387.7 kB view details)

Uploaded CPython 3.7mmacOS 10.9+ x86-64

torchvision-0.7.0-cp36-cp36m-manylinux1_x86_64.whl (5.9 MB view details)

Uploaded CPython 3.6m

torchvision-0.7.0-cp36-cp36m-macosx_10_9_x86_64.whl (387.7 kB view details)

Uploaded CPython 3.6mmacOS 10.9+ x86-64

File details

Details for the file torchvision-0.7.0-cp38-cp38-manylinux1_x86_64.whl.

File metadata

  • Download URL: torchvision-0.7.0-cp38-cp38-manylinux1_x86_64.whl
  • Upload date:
  • Size: 5.9 MB
  • Tags: CPython 3.8
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.1.1 pkginfo/1.5.0.1 requests/2.24.0 setuptools/47.3.1.post20200622 requests-toolbelt/0.9.1 tqdm/4.47.0 CPython/3.8.1

File hashes

Hashes for torchvision-0.7.0-cp38-cp38-manylinux1_x86_64.whl
Algorithm Hash digest
SHA256 cfa2b367bc9acf20f18b151d0525970279719e81969c17214effe77245875354
MD5 d71cba052f857350dc799592cb9c1a96
BLAKE2b-256 38a7a6a009991d0df86ef76023f1334a0d1201b310dd0e7abb4d3e64b080cc87

See more details on using hashes here.

File details

Details for the file torchvision-0.7.0-cp38-cp38-macosx_10_9_x86_64.whl.

File metadata

  • Download URL: torchvision-0.7.0-cp38-cp38-macosx_10_9_x86_64.whl
  • Upload date:
  • Size: 387.7 kB
  • Tags: CPython 3.8, macOS 10.9+ x86-64
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.1.1 pkginfo/1.5.0.1 requests/2.24.0 setuptools/47.3.1.post20200622 requests-toolbelt/0.9.1 tqdm/4.47.0 CPython/3.8.1

File hashes

Hashes for torchvision-0.7.0-cp38-cp38-macosx_10_9_x86_64.whl
Algorithm Hash digest
SHA256 f5686e0a0dd511ac33eb9d6279bd34edd9f282dcb7c8ad21e290882c6206504f
MD5 391c1712447607d6437309c53b96fda9
BLAKE2b-256 f4896e6e508781858299e74dacf157d46853701348243ba2f29eee37ec5fcf73

See more details on using hashes here.

File details

Details for the file torchvision-0.7.0-cp37-cp37m-manylinux1_x86_64.whl.

File metadata

  • Download URL: torchvision-0.7.0-cp37-cp37m-manylinux1_x86_64.whl
  • Upload date:
  • Size: 5.9 MB
  • Tags: CPython 3.7m
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.1.1 pkginfo/1.5.0.1 requests/2.24.0 setuptools/47.3.1.post20200622 requests-toolbelt/0.9.1 tqdm/4.47.0 CPython/3.8.1

File hashes

Hashes for torchvision-0.7.0-cp37-cp37m-manylinux1_x86_64.whl
Algorithm Hash digest
SHA256 0d1a5adfef4387659c7a0af3b72e16caa0c67224a422050ab65184d13ac9fb13
MD5 a3822d7c1b1a32db50850ad1c7cea4a2
BLAKE2b-256 4db560d5eb61f1880707a5749fea43e0ec76f27dfe69391cdec953ab5da5e676

See more details on using hashes here.

File details

Details for the file torchvision-0.7.0-cp37-cp37m-macosx_10_9_x86_64.whl.

File metadata

  • Download URL: torchvision-0.7.0-cp37-cp37m-macosx_10_9_x86_64.whl
  • Upload date:
  • Size: 387.7 kB
  • Tags: CPython 3.7m, macOS 10.9+ x86-64
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.1.1 pkginfo/1.5.0.1 requests/2.24.0 setuptools/47.3.1.post20200622 requests-toolbelt/0.9.1 tqdm/4.47.0 CPython/3.8.1

File hashes

Hashes for torchvision-0.7.0-cp37-cp37m-macosx_10_9_x86_64.whl
Algorithm Hash digest
SHA256 8c8df7e1d1f3d4e088256be1e8c8d3eb90b016302baa4649742d47ae1531da37
MD5 77f5821ad29b715ff0e22bfea72fe2a7
BLAKE2b-256 7268b82d188d09a40e681e8df5eeb91f71bb1facdc5c8a61a905350fa398a4a4

See more details on using hashes here.

File details

Details for the file torchvision-0.7.0-cp36-cp36m-manylinux1_x86_64.whl.

File metadata

  • Download URL: torchvision-0.7.0-cp36-cp36m-manylinux1_x86_64.whl
  • Upload date:
  • Size: 5.9 MB
  • Tags: CPython 3.6m
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.1.1 pkginfo/1.5.0.1 requests/2.24.0 setuptools/47.3.1.post20200622 requests-toolbelt/0.9.1 tqdm/4.47.0 CPython/3.8.1

File hashes

Hashes for torchvision-0.7.0-cp36-cp36m-manylinux1_x86_64.whl
Algorithm Hash digest
SHA256 14c0bf60fa26aabaea64ef30b8e5d441ee78d1a5eed568c30806af19bbe6b638
MD5 e674a5f3925238ba1056139d1a0ab93c
BLAKE2b-256 8edc4a939cfbd38398f4765f712576df21425241020bfccc200af76d19088533

See more details on using hashes here.

File details

Details for the file torchvision-0.7.0-cp36-cp36m-macosx_10_9_x86_64.whl.

File metadata

  • Download URL: torchvision-0.7.0-cp36-cp36m-macosx_10_9_x86_64.whl
  • Upload date:
  • Size: 387.7 kB
  • Tags: CPython 3.6m, macOS 10.9+ x86-64
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/3.1.1 pkginfo/1.5.0.1 requests/2.24.0 setuptools/47.3.1.post20200622 requests-toolbelt/0.9.1 tqdm/4.47.0 CPython/3.8.1

File hashes

Hashes for torchvision-0.7.0-cp36-cp36m-macosx_10_9_x86_64.whl
Algorithm Hash digest
SHA256 a70d80bb8749c1e4a46fa56dc2fc857e98d14600841e02cc2fed766daf96c245
MD5 031111de5fe4116f5ed15a114aeaf4b3
BLAKE2b-256 c69d1d2aa44d47d2a3c1bb5620a15cc613f401e438fc6bd0afeff424ca520f84

See more details on using hashes here.

Supported by

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