pytorch/vision

★ 17,912⑂ 0

Datasets, Transforms and Models specific to Computer Vision

About pytorch/vision

pytorch/vision is an open-source project on GitHub, mainly written in Python. Datasets, Transforms and Models specific to Computer Vision It currently holds 17,912 stars and 0 forks with 0 open issues, and was last pushed on an unknown date (repository created unknown).

Project Overview

AI Homed tracks it on the AI Image Projects board and on the AI AI Image Projects list.

GitHub Repository Details

Repository pytorch/vision · default branch - · size 0 KB · watchers 0 · source: GitHub REST API and repository README

README

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:

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

GitHub Stars & Activity

17,912Stars
0Forks
0Open issues
PythonLanguage

GitHub Popularity

GitHub stars17,912
Forks0
Open issues0
Primary languagePython
License-
Stars gained today0
Created-
Last pushed-

Trending History

Trending statusnot on today's boards

Related AI Projects

1

d2l-ai / d2l-zh

Python★ 80,716⑂ 0
2

ultralytics / ultralytics

Python★ 61,651⑂ 0
3

ultralytics / yolov5

Python★ 58,015⑂ 0
4
5

roboflow / supervision

Python★ 50,364⑂ 0
6

d2l-ai / d2l-en

Python★ 29,615⑂ 0
7

lucidrains / vit-pytorch

Python★ 25,506⑂ 0
8

More AI Rankings