Skip to main content

image and video datasets and models for torch deep learning

Project description

torchvision

total torchvision downloads documentation

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

Installation

Please refer to the official instructions to install the stable versions of torch and torchvision on your system.

To build source, refer to our contributing page.

The following is the corresponding torchvision versions and supported Python versions.

torch torchvision Python
main / nightly main / nightly >=3.9, <=3.12
2.4 0.19 >=3.8, <=3.12
2.3 0.18 >=3.8, <=3.12
2.2 0.17 >=3.8, <=3.11
2.1 0.16 >=3.8, <=3.11
2.0 0.15 >=3.8, <=3.11
older versions
torch torchvision Python
1.13 0.14 >=3.7.2, <=3.10
1.12 0.13 >=3.7, <=3.10
1.11 0.12 >=3.7, <=3.10
1.10 0.11 >=3.6, <=3.9
1.9 0.10 >=3.6, <=3.9
1.8 0.9 >=3.6, <=3.9
1.7 0.8 >=3.6, <=3.9
1.6 0.7 >=3.6, <=3.8
1.5 0.6 >=3.5, <=3.8
1.4 0.5 ==2.7, >=3.5, <=3.8
1.3 0.4.2 / 0.4.3 ==2.7, >=3.5, <=3.7
1.2 0.4.1 ==2.7, >=3.5, <=3.7
1.1 0.3 ==2.7, >=3.5, <=3.7
<=1.0 0.2 ==2.7, >=3.5, <=3.7

Image Backends

Torchvision currently supports the following image backends:

  • torch tensors
  • PIL images:

Read more in in our docs.

[UNSTABLE] Video Backend

Torchvision currently supports the following video backends:

  • pyav (default) - Pythonic binding for ffmpeg libraries.
  • video_reader - This needs ffmpeg to be installed and torchvision to be built from source. There shouldn't be any conflicting version of ffmpeg installed. Currently, this is only supported on Linux.
conda install -c conda-forge 'ffmpeg<4.3'
python setup.py install

Using the models on C++

Refer to example/cpp.

DISCLAIMER: the libtorchvision library includes the torchvision custom ops as well as most of the C++ torchvision APIs. Those APIs do not come with any backward-compatibility guarantees and may change from one version to the next. Only the Python APIs are stable and with backward-compatibility guarantees. So, if you need stability within a C++ environment, your best bet is to export the Python APIs via torchscript.

Documentation

You can find the API documentation on the pytorch website: https://pytorch.org/vision/stable/index.html

Contributing

See the CONTRIBUTING file for how to help out.

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!

Pre-trained Model License

The pre-trained models provided in this library may have their own licenses or terms and conditions derived from the dataset used for training. It is your responsibility to determine whether you have permission to use the models for your use case.

More specifically, SWAG models are released under the CC-BY-NC 4.0 license. See SWAG LICENSE for additional details.

Citing TorchVision

If you find TorchVision useful in your work, please consider citing the following BibTeX entry:

@software{torchvision2016,
    title        = {TorchVision: PyTorch's Computer Vision library},
    author       = {TorchVision maintainers and contributors},
    year         = 2016,
    journal      = {GitHub repository},
    publisher    = {GitHub},
    howpublished = {\url{https://github.com/pytorch/vision}}
}

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.20.1-cp312-cp312-win_amd64.whl (1.6 MB view details)

Uploaded CPython 3.12 Windows x86-64

torchvision-0.20.1-cp312-cp312-manylinux1_x86_64.whl (7.2 MB view details)

Uploaded CPython 3.12

torchvision-0.20.1-cp312-cp312-macosx_11_0_arm64.whl (1.8 MB view details)

Uploaded CPython 3.12 macOS 11.0+ ARM64

torchvision-0.20.1-cp311-cp311-win_amd64.whl (1.6 MB view details)

Uploaded CPython 3.11 Windows x86-64

torchvision-0.20.1-cp311-cp311-manylinux1_x86_64.whl (7.2 MB view details)

Uploaded CPython 3.11

torchvision-0.20.1-cp311-cp311-macosx_11_0_arm64.whl (1.8 MB view details)

Uploaded CPython 3.11 macOS 11.0+ ARM64

torchvision-0.20.1-cp310-cp310-win_amd64.whl (1.6 MB view details)

Uploaded CPython 3.10 Windows x86-64

torchvision-0.20.1-cp310-cp310-manylinux1_x86_64.whl (7.2 MB view details)

Uploaded CPython 3.10

torchvision-0.20.1-cp310-cp310-macosx_11_0_arm64.whl (1.8 MB view details)

Uploaded CPython 3.10 macOS 11.0+ ARM64

torchvision-0.20.1-cp39-cp39-win_amd64.whl (1.6 MB view details)

Uploaded CPython 3.9 Windows x86-64

torchvision-0.20.1-cp39-cp39-manylinux1_x86_64.whl (7.2 MB view details)

Uploaded CPython 3.9

torchvision-0.20.1-cp39-cp39-macosx_11_0_arm64.whl (1.8 MB view details)

Uploaded CPython 3.9 macOS 11.0+ ARM64

File details

Details for the file torchvision-0.20.1-cp312-cp312-win_amd64.whl.

File metadata

File hashes

Hashes for torchvision-0.20.1-cp312-cp312-win_amd64.whl
Algorithm Hash digest
SHA256 4a330422c36dbfc946d3a6c1caec3489db07ecdf3675d83369adb2e5a0ca17c4
MD5 3312d4b4321656621af58f77ae8a4df8
BLAKE2b-256 c31800993d420b1d6e88582e51d4bc82c824c99a2e9c045d50eaf9b34fff729a

See more details on using hashes here.

File details

Details for the file torchvision-0.20.1-cp312-cp312-manylinux2014_aarch64.whl.

File metadata

File hashes

Hashes for torchvision-0.20.1-cp312-cp312-manylinux2014_aarch64.whl
Algorithm Hash digest
SHA256 9f853ba4497ac4691815ad41b523ee23cf5ba4f87b1ce869d704052e233ca8b7
MD5 e462cd8511fead7e71dcd25390afd489
BLAKE2b-256 aff0ca1445406eb12cbeb7a41fc833a1941ede78e7c55621198b83ecd7bcfd0f

See more details on using hashes here.

File details

Details for the file torchvision-0.20.1-cp312-cp312-manylinux1_x86_64.whl.

File metadata

File hashes

Hashes for torchvision-0.20.1-cp312-cp312-manylinux1_x86_64.whl
Algorithm Hash digest
SHA256 17cd78adddf81dac57d7dccc9277a4d686425b1c55715f308769770cb26cad5c
MD5 3362eeb10feb923a6bff7440e6120cb3
BLAKE2b-256 d47500a852275ade58d3dc474530f7a7b6bc999a817148f0eb59d4fde12eb955

See more details on using hashes here.

File details

Details for the file torchvision-0.20.1-cp312-cp312-macosx_11_0_arm64.whl.

File metadata

File hashes

Hashes for torchvision-0.20.1-cp312-cp312-macosx_11_0_arm64.whl
Algorithm Hash digest
SHA256 1a31256ff945d64f006bb306813a7c95a531fe16bfb2535c837dd4c104533d7a
MD5 f2e49c3b81664c24886524fe9428e700
BLAKE2b-256 c5eb4ba19616378f2bc085999432fded2b7dfdbdccc6dd0fc293203452508100

See more details on using hashes here.

File details

Details for the file torchvision-0.20.1-cp311-cp311-win_amd64.whl.

File metadata

File hashes

Hashes for torchvision-0.20.1-cp311-cp311-win_amd64.whl
Algorithm Hash digest
SHA256 5b501d5c04b034d2ecda96a31ed050e383cf8201352e4c9276ca249cbecfded0
MD5 a2c4c8c8fbfdbe3d56851d62b13aafc3
BLAKE2b-256 6955ce836703ff77bb21582c3098d5311f8ddde7eadc7eab04be9561961f4725

See more details on using hashes here.

File details

Details for the file torchvision-0.20.1-cp311-cp311-manylinux2014_aarch64.whl.

File metadata

File hashes

Hashes for torchvision-0.20.1-cp311-cp311-manylinux2014_aarch64.whl
Algorithm Hash digest
SHA256 a40d766345927639da322c693934e5f91b1ba2218846c7104b868dea2314ce8e
MD5 fedbf602bad4d4f63bb65144ae5b1a4f
BLAKE2b-256 b1a3cbb8177e5e379f0c040b00c6f80f14d323a97e30495d7115d169b101b2f7

See more details on using hashes here.

File details

Details for the file torchvision-0.20.1-cp311-cp311-manylinux1_x86_64.whl.

File metadata

File hashes

Hashes for torchvision-0.20.1-cp311-cp311-manylinux1_x86_64.whl
Algorithm Hash digest
SHA256 86f6523dee420000fe14c3527f6c8e0175139fda7d995b187f54a0b0ebec7eb6
MD5 2dd12cda7791151268bb6ff66a1aef1c
BLAKE2b-256 dee9e190ecec448d5a2abad8348cf085fcb39962a491e3f40dcb023721e04feb

See more details on using hashes here.

File details

Details for the file torchvision-0.20.1-cp311-cp311-macosx_11_0_arm64.whl.

File metadata

File hashes

Hashes for torchvision-0.20.1-cp311-cp311-macosx_11_0_arm64.whl
Algorithm Hash digest
SHA256 344b339e15e6bbb59ee0700772616d0afefd209920c762b1604368d8c3458322
MD5 49dc0eba2266215e036b0233589dd7fa
BLAKE2b-256 28574d7ad90be612f5ac6c4bdafcb0ff13e818e14a340a88c8ca00d9ed8c2dad

See more details on using hashes here.

File details

Details for the file torchvision-0.20.1-cp310-cp310-win_amd64.whl.

File metadata

File hashes

Hashes for torchvision-0.20.1-cp310-cp310-win_amd64.whl
Algorithm Hash digest
SHA256 22c2fa44e20eb404b85e42b22b453863a14b0927d25e550fd4f84eea97fa5b39
MD5 7ce5006c4ac6b28ce23929f1897b5df8
BLAKE2b-256 1711b5ce67715bbbec8798fb48c4a20ac28828aec1710ac01091a3eddcb8e075

See more details on using hashes here.

File details

Details for the file torchvision-0.20.1-cp310-cp310-manylinux2014_aarch64.whl.

File metadata

File hashes

Hashes for torchvision-0.20.1-cp310-cp310-manylinux2014_aarch64.whl
Algorithm Hash digest
SHA256 75f8a4d51a593c4bab6c9bf7d75bdd88691b00a53b07656678bc55a3a753dd73
MD5 e8707a957eebc7fc36aba131a5703932
BLAKE2b-256 f7ce4c31e9b96cc4f9fec746b258d2aa35f8d1247f4f58d63f9c505ea5eb254d

See more details on using hashes here.

File details

Details for the file torchvision-0.20.1-cp310-cp310-manylinux1_x86_64.whl.

File metadata

File hashes

Hashes for torchvision-0.20.1-cp310-cp310-manylinux1_x86_64.whl
Algorithm Hash digest
SHA256 8ffbdf8bf5b30eade22d459f5a313329eeadb20dc75efa142987b53c007098c3
MD5 c9fd1a852bbc6e495bd9081370777dde
BLAKE2b-256 a2f67ff89a9f8703f623f5664afd66c8600e3f09fe188e1e0b7e6f9a8617f865

See more details on using hashes here.

File details

Details for the file torchvision-0.20.1-cp310-cp310-macosx_11_0_arm64.whl.

File metadata

File hashes

Hashes for torchvision-0.20.1-cp310-cp310-macosx_11_0_arm64.whl
Algorithm Hash digest
SHA256 4878fefb96ef293d06c27210918adc83c399d9faaf34cda5a63e129f772328f1
MD5 998f98684f8e9f73d40d133b406ea062
BLAKE2b-256 8d59aea68d755da1451e1a0d894528a7edc9b58eb30d33e274bf21bef28dad1a

See more details on using hashes here.

File details

Details for the file torchvision-0.20.1-cp39-cp39-win_amd64.whl.

File metadata

File hashes

Hashes for torchvision-0.20.1-cp39-cp39-win_amd64.whl
Algorithm Hash digest
SHA256 ea9678163bbf19568f4f959d927f3751eeb833cc8eac949de507edde38c1fc9f
MD5 8655df3fdecb07e38aedcfb3e644684c
BLAKE2b-256 d41e0bd619dd8aa443e167fb62c6167ecb9b6ce8f275e9ca1842e7994fd9653d

See more details on using hashes here.

File details

Details for the file torchvision-0.20.1-cp39-cp39-manylinux2014_aarch64.whl.

File metadata

File hashes

Hashes for torchvision-0.20.1-cp39-cp39-manylinux2014_aarch64.whl
Algorithm Hash digest
SHA256 abcb8005de8dc393dbd1310ecb669dc68ab664b9107af6d698a6341d1d3f2c3c
MD5 556ff6774be9a71e752d5627f2d3a076
BLAKE2b-256 44a7a69e090ee59b6b042580304524af035dc0e980f05f9b0592f2153de94579

See more details on using hashes here.

File details

Details for the file torchvision-0.20.1-cp39-cp39-manylinux1_x86_64.whl.

File metadata

File hashes

Hashes for torchvision-0.20.1-cp39-cp39-manylinux1_x86_64.whl
Algorithm Hash digest
SHA256 408766b2f0ada9e1bc880d12346cec9638535af5df6459ba9ac4ce5c46402f91
MD5 f6162fcf1ad6e554030ea9d841398d6d
BLAKE2b-256 f16cf5fcf2e2a5828643354939ececbcd45bfde4d0355b1e44722c960c6f81c2

See more details on using hashes here.

File details

Details for the file torchvision-0.20.1-cp39-cp39-macosx_11_0_arm64.whl.

File metadata

File hashes

Hashes for torchvision-0.20.1-cp39-cp39-macosx_11_0_arm64.whl
Algorithm Hash digest
SHA256 2cd58406978b813188cf4e9135b218775b57e0bb86d4a88f0339874b8a224819
MD5 3f42e2f0190a8c07b11a51efd2825c81
BLAKE2b-256 a114c13e8b49fa812266e3340969b3157b11928a3608faa3a0448b8a564b01b8

See more details on using hashes here.

Supported by

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