torch-vision
This repository consists of:
vision.datasets
: Data loaders for popular vision datasetsvision.transforms
: Common image transformations such as random crop, rotations etc.[WIP] vision.models
: Model definitions and Pre-trained models for popular models such as AlexNet, VGG, ResNet etc.
Installation
Binaries:
conda install pytorch-vision -c https://conda.anaconda.org/t/6N-MsQ4WZ7jo/soumith
From Source:
pip install -r requirements.txt
pip install .