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

Anaconda:

conda install torchvision -c soumith

pip:

pip install torchvision

From source:

python setup.py install

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

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.