image and video datasets and models for torch deep learning
The torchvision package consists of popular datasets, model architectures, and common image transformations for computer vision.
conda install torchvision -c pytorch
pip install torchvision
python setup.py install
Torchvision currently supports the following image backends:
You can find the API documentation 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.