跳转到主要内容

torch深度学习用图像和视频数据集和模型

项目描述

torchvision

total torchvision downloads documentation

torchvision 包包含计算机视觉领域流行的数据集、模型架构和常见图像变换。

安装

请参考官方说明,在您的系统上安装 torchtorchvision 的稳定版本。

要构建源代码,请参考我们的贡献页面

以下是相应的 torchvision 版本和支持的 Python 版本。

torch torchvision Python
main / nightly main / nightly >=3.8, <=3.12
2.3 0.18 >=3.8, <=3.12
2.2 0.17 >=3.8, <=3.11
2.1 0.16 >=3.8, <=3.11
2.0 0.15 >=3.8, <=3.11
旧版本
torch torchvision Python
1.13 0.14 >=3.7.2, <=3.10
1.12 0.13 >=3.7, <=3.10
1.11 0.12 >=3.7, <=3.10
1.10 0.11 >=3.6, <=3.9
1.9 0.10 >=3.6, <=3.9
1.8 0.9 >=3.6, <=3.9
1.7 0.8 >=3.6, <=3.9
1.6 0.7 >=3.6, <=3.8
1.5 0.6 >=3.5, <=3.8
1.4 0.5 ==2.7, >=3.5, <=3.8
1.3 0.4.2 / 0.4.3 ==2.7, >=3.5, <=3.7
1.2 0.4.1 ==2.7, >=3.5, <=3.7
1.1 0.3 ==2.7, >=3.5, <=3.7
<=1.0 0.2 ==2.7, >=3.5, <=3.7

图像后端

torchvision 当前支持以下图像后端

  • torch 矩阵
  • PIL 图像

更多详情请参阅我们的文档

[不稳定] 视频后端

torchvision 当前支持以下视频后端

  • pyav(默认)- ffmpeg 库的 Python 绑定。
  • video_reader - 这需要安装 ffmpeg 并且从源代码构建 torchvision。不应安装任何冲突的 ffmpeg 版本。目前,这仅在 Linux 上支持。
conda install -c conda-forge 'ffmpeg<4.3'
python setup.py install

在 C++ 中使用模型

请参考示例/cpp

免责声明:`libtorchvision` 库包含 torchvision 自定义操作以及大多数 C++ torchvision API。这些 API 不提供向后兼容性保证,并且可能会随版本变化。只有 Python API 是稳定的,并具有向后兼容性保证。因此,如果您需要在 C++ 环境中保持稳定性,最好的方法是使用 torchscript 导出 Python API。

文档

您可以在 PyTorch 网站上找到 API 文档:https://pytorch.ac.cn/vision/stable/index.html

贡献

有关如何帮助的详细信息,请参阅CONTRIBUTING 文件。

关于数据集的免责声明

这是一个实用程序库,用于下载和准备公共数据集。我们不托管或分发这些数据集,也不对其质量或公平性作出保证,也不声称您有权使用数据集。确定您是否可以使用数据集的许可权是您的责任。

如果您是数据集的所有者并希望更新其任何部分(描述、引用等),或者不希望您的数据集包含在此库中,请通过 GitHub 的问题与我们联系。感谢您对机器学习社区的贡献!

预训练模型许可

此库中提供的预训练模型可能有自己的许可或条款和条件,这些许可或条款和条件是从用于训练的数据集中派生的。您有责任确定您是否有权根据数据集的许可使用模型。

具体来说,SWAG 模型是在 CC-BY-NC 4.0 许可下发布的。有关更多详细信息,请参阅SWAG 许可证

引用 TorchVision

如果您在工作中发现 TorchVision 很有用,请考虑引用以下 BibTeX 条目

@software{torchvision2016,
    title        = {TorchVision: PyTorch's Computer Vision library},
    author       = {TorchVision maintainers and contributors},
    year         = 2016,
    journal      = {GitHub repository},
    publisher    = {GitHub},
    howpublished = {\url{https://github.com/pytorch/vision}}
}

项目详情


发布历史 发布通知 | RSS 源

下载文件

下载适用于您平台的文件。如果您不确定选择哪一个,请了解更多关于安装包的信息。

源代码分发

此版本没有提供源代码分发文件。请参阅生成分发存档的教程。

构建分发

torchvision-0.19.1-cp312-cp312-win_amd64.whl (1.3 MB 查看哈希值)

上传日期 CPython 3.12 Windows x86-64

torchvision-0.19.1-cp312-cp312-manylinux2014_aarch64.whl (14.1 MB 查看哈希值)

上传日期 CPython 3.12

torchvision-0.19.1-cp312-cp312-manylinux1_x86_64.whl (7.0 MB 查看哈希值)

上传日期 CPython 3.12

torchvision-0.19.1-cp312-cp312-macosx_11_0_arm64.whl (1.7 MB 查看哈希值)

上传日期 CPython 3.12 macOS 11.0+ ARM64

torchvision-0.19.1-cp311-cp311-win_amd64.whl (1.3 MB 查看哈希值)

上传日期 CPython 3.11 Windows x86-64

torchvision-0.19.1-cp311-cp311-manylinux2014_aarch64.whl (14.1 MB 查看哈希值)

上传日期 CPython 3.11

torchvision-0.19.1-cp311-cp311-manylinux1_x86_64.whl (7.0 MB 查看哈希值)

上传时间: CPython 3.11

torchvision-0.19.1-cp311-cp311-macosx_11_0_arm64.whl (1.7 MB 查看哈希值)

上传时间: CPython 3.11 macOS 11.0+ ARM64

torchvision-0.19.1-cp310-cp310-win_amd64.whl (1.3 MB 查看哈希值)

上传时间: CPython 3.10 Windows x86-64

torchvision-0.19.1-cp310-cp310-manylinux2014_aarch64.whl (14.1 MB 查看哈希值)

上传时间: CPython 3.10

torchvision-0.19.1-cp310-cp310-manylinux1_x86_64.whl (7.0 MB 查看哈希值)

上传时间: CPython 3.10

torchvision-0.19.1-cp310-cp310-macosx_11_0_arm64.whl (1.7 MB 查看哈希值)

上传时间: CPython 3.10 macOS 11.0+ ARM64

torchvision-0.19.1-cp39-cp39-win_amd64.whl (1.3 MB 查看哈希值)

上传时间: CPython 3.9 Windows x86-64

torchvision-0.19.1-cp39-cp39-manylinux2014_aarch64.whl (1.7 MB 查看哈希值)

上传时间: CPython 3.9

torchvision-0.19.1-cp39-cp39-manylinux1_x86_64.whl (7.0 MB 查看哈希值)

上传时间: CPython 3.9

torchvision-0.19.1-cp39-cp39-macosx_11_0_arm64.whl (1.7 MB 查看哈希值)

上传时间: CPython 3.9 macOS 11.0+ ARM64

torchvision-0.19.1-cp38-cp38-win_amd64.whl (1.3 MB 查看哈希值)

上传时间: CPython 3.8 Windows x86-64

torchvision-0.19.1-cp38-cp38-manylinux2014_aarch64.whl (14.1 MB 查看哈希值)

上传时间 CPython 3.8

torchvision-0.19.1-cp38-cp38-manylinux1_x86_64.whl (7.0 MB 查看哈希值)

上传时间 CPython 3.8

torchvision-0.19.1-cp38-cp38-macosx_11_0_arm64.whl (1.7 MB 查看哈希值)

上传时间 CPython 3.8 macOS 11.0+ ARM64

由以下机构支持

AWS AWS 云计算和安全赞助商 Datadog Datadog 监控 Fastly Fastly CDN Google Google 下载分析 Microsoft Microsoft PSF 赞助商 Pingdom Pingdom 监控 Sentry Sentry 错误日志 StatusPage StatusPage 状态页面