torchvision 0.4.0

火炬视野

TorchVision软件包包括流行的数据集、模型结构和用于计算机视觉的常见图像转换。

安装

torchvision需要pytorch 1.1或更新版本。

水蟒：

conda install torchvision -c pytorch

管道：

pip install torchvision

来源：

python setup.py install
# or, for OSX
# MACOSX_DEPLOYMENT_TARGET=10.9 CC=clang CXX=clang++ python setup.py install

默认情况下，如果找到cuda并且torch.cuda.is_available()为true，则生成gpu支持。 可以通过设置FORCE_CUDA=1环境变量来强制构建GPU支持， 这在构建Docker图像时很有用。

图像后端

torchvision目前支持以下图像后端：

• Pillow（默认）
• Pillow-SIMD-a更快用simd替换枕头。如果已安装，将用作默认设置。
• accimage-如果已安装，则可以通过调用torchvision.set_image_backend('accimage')

C++API

TooVistVIEW还提供了一个C++的API，其中包含了相当于Python模型的C++。

从源安装：

mkdir build
cd build
cmake ..
make
make install

文档

您可以在pytorch网站上找到api文档：http://pytorch.org/docs/master/torchvision/

贡献

我们感谢所有的贡献。如果你打算贡献回错误修复，请这样做，没有任何进一步的讨论。如果您计划提供新功能、实用程序功能或扩展，请首先打开一个问题并与我们讨论该功能。