README.rst 899 Bytes
Newer Older
Thomas Grainger's avatar
Thomas Grainger committed
1
2
3
torch-vision
============

4
5
6
.. image:: https://travis-ci.org/pytorch/vision.svg?branch=master
    :target: https://travis-ci.org/pytorch/vision

7
The torchvision package consists of popular datasets, model architectures, and common image transformations for computer vision.
Thomas Grainger's avatar
Thomas Grainger committed
8
9
10
11

Installation
============

Soumith Chintala's avatar
Soumith Chintala committed
12
Anaconda:
Thomas Grainger's avatar
Thomas Grainger committed
13
14
15

.. code:: bash

Soumith Chintala's avatar
Soumith Chintala committed
16
    conda install torchvision -c soumith
Thomas Grainger's avatar
Thomas Grainger committed
17

Soumith Chintala's avatar
Soumith Chintala committed
18
pip:
Thomas Grainger's avatar
Thomas Grainger committed
19
20
21

.. code:: bash

Thomas Grainger's avatar
Thomas Grainger committed
22
    pip install torchvision
Thomas Grainger's avatar
Thomas Grainger committed
23

Soumith Chintala's avatar
Soumith Chintala committed
24
25
26
27
28
29
From source:

.. code:: bash

    python setup.py install

30
31
32
Documentation
=============
You can find the API doucmentation on the pytorch website: http://pytorch.org/docs/master/torchvision/
edgarriba's avatar
edgarriba committed
33

34
35
36
Contributing
============
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.