vision README

The project's README from github.com/pytorch/vision. As one markdown file: README.md.

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.10, <=3.14
2.13 0.28 >=3.10, <=3.14
2.12 0.27 >=3.10, <=3.14
2.11 0.26 >=3.10, <=3.14
2.10 0.25 >=3.10, <=3.14
older versions
torch torchvision Python
2.9 0.24 >=3.10, <=3.14
2.8 0.23 >=3.9, <=3.13
2.7 0.22 >=3.9, <=3.13
2.6 0.21 >=3.9, <=3.12
2.5 0.20 >=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
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:

Read more in in our docs.

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}}
}

vision (pytorch/vision)

Searching Database…

18K ·GitHub
Что это такое?Torchvision — это пакет PyTorch для компьютерного зрения: популярные датасеты, готовые архитектуры моделей и стандартные преобразования изображений для задач со зрением.
Это отдельный фреймворк или часть PyTorch?Это сопутствующая библиотека, а не фреймворк: она устанавливается вместе с PyTorch с совпадающими версиями (torchvision 0.24 соответствует torch 2.9) и строится на тензорной экосистеме torch.
Нужна ли какая-то конкретная версия Python или torch?Да — версии жёстко привязаны: каждая версия torchvision соответствует определённой версии torch и диапазону Python, например 0.24 идёт с torch 2.9, а сборки main/nightly требуют Python 3.10+.
Хранит ли torchvision датасеты, к которым даёт доступ?Нет — библиотека только скачивает и подготавливает публичные датасеты. Проверять, разрешено ли вам использовать датасет по его лицензии, нужно полностью самостоятельно.
Лицензия свободная или есть подводные камни?Код распространяется по BSD-3-Clause, но у предобученных моделей могут быть свои условия от данных, на которых они обучались, — например, веса SWAG под лицензией CC-BY-NC 4.0, только для некоммерческого использования.
Как выглядит типичный рабочий процесс с ней?Загрузите публичный датасет, прогоните изображения через конвейер преобразований для предобработки и аугментации, затем обучите или дообучите (fine-tune) одну из включённых предобученных архитектур моделей.
Какие форматы изображений она принимает?Работает с тензорами torch и изображениями PIL. Для более быстрого декодирования можно заменить Pillow на Pillow-SIMD — drop-in-замену, ускоренную с помощью SIMD.
Как сослаться на неё в научной работе?В README есть запись BibTeX для цитирования TorchVision в академических работах, а полная документация по API размещена на сайте PyTorch.
Проект всё ещё активно развивается?Да — он следует за релизами PyTorch с опубликованной таблицей совместимости (torchvision 0.24 для torch 2.9), выходят nightly-сборки, у репозитория около 18 тысяч звёзд, и он не архивирован.
Creator
pytorch
Size
1.2 GB
Stars
18K
Age
9y
Commits · 90 Days