vision:基于昇腾NPU的Torchvision框架适配项目

本项目开发了Torchvision Adapter插件,用于昇腾适配Torchvision框架。 目前该适配框架增加了对Torchvision所提供的常用算子的支持,提供了基于cv2和基于昇腾NPU的图像处理加速后端以加速图像处理。

分支6Tags11
文件最后提交记录最后更新时间
!132 add ci for vision1 年前
modify document15 小时前
!164 Support pillow Image format10 个月前
Add dvpp .h file6 个月前
fix video_backend19 天前
!71 update torchvision for 0.16.02 年前
!71 update torchvision for 0.16.02 年前
!71 update torchvision for 0.16.02 年前
Update URL (gitcode)8 个月前
modify document1 个月前
!161 Update third party and fix PathManager warning bug11 个月前
update requirement.txt2 个月前
Update URL (gitcode)8 个月前
Add dvpp .h file6 个月前
!173 vision support 0.22.19 个月前

Torchvision Adapter

简介

本项目开发了Torchvision Adapter插件,用于昇腾适配Torchvision框架。 目前该适配框架增加了对Torchvision所提供的常用算子的支持,提供了基于cv2和基于昇腾NPU的图像处理加速后端以加速图像处理。

目录结构

关键目录如下:

├─ci                             # 持续集成脚本目录
├─torchvision_npu                # 核心适配目录
│  ├─csrc/                       # 底层核心目录
│  ├─datasets/                   # 数据集加载适配目录
│  ├─io/                         # 图像io加速目录
│  ├─transforms/                 # 图像预处理核心目录
│  ├─ops/                        # 视觉算子NPU实现目录
├─docs                           # 项目文档目录
├─third_party/                   # 外部依赖目录
└─test                           # 测试目录

版本说明

Torchvision Adapter分支 Torchvision Adapter Tag PyTorch版本 PyTorch Extension版本 Torchvision版本 驱动版本 CANN版本
master - 2.7.1 2.7.1.post2 0.22.1 Ascend HDK 25.3.0 CANN 8.5.0
v0.21.0-dev v0.21.0-7.1.0 2.6.0 2.6.0 0.21.0 Ascend HDK 25.2.0 CANN 8.2.RC1
v0.20.1-dev v0.20.1-7.1.0 2.5.1 2.5.1.post1 0.20.1 Ascend HDK 25.2.0 CANN 8.2.RC1
v0.16.0-dev v0.16.0-7.1.0 2.1.0 2.1.0.post13 0.16.0 Ascend HDK 25.2.0 CANN 8.2.RC1
v0.20.1-dev v0.20.1-7.0.0 2.5.1 2.5.1 0.20.1 Ascend HDK 25.0.RC1 CANN 8.1.RC1
v0.16.0-dev v0.16.0-7.0.0 2.1.0 2.1.0.post12 0.16.0 Ascend HDK 25.0.RC1 CANN 8.1.RC1
v0.16.0-dev v0.16.0-6.0.0 2.1.0 2.1.0.post10 0.16.0 Ascend HDK 24.1.RC3 CANN 8.0.0
v0.16.0-dev v0.16.0-6.0.rc3 2.1.0 2.1.0.post8 0.16.0 Ascend HDK 24.1.RC3 CANN 8.0.RC3
v0.16.0-dev v0.16.0-6.0.rc2 2.1.0 2.1.0.post6 0.16.0 Ascend HDK 24.1.RC2 CANN 8.0.RC2
v0.12.0-dev v0.12.0-6.0.rc2 1.11.0 1.11.0.post14 0.12.0 Ascend HDK 24.1.RC2 CANN 8.0.RC2

环境部署

Torchvision Adapter插件的安装操作,具体请参见《Torchvision Adapter插件 软件安装》。

快速入门

Torchvision Adapter插件的快速入门,具体请参见《Torchvision Adapter插件 快速入门》。

特性介绍

Torchvision Adapter包括如下主要特性,各特性用法请参见《Torchvision Adapter特性指南》:

  • 使用cv2图像处理后端
  • 使用DVPP图像处理后端
  • 使用DVPP视频处理后端
  • 数据预处理使用DVPP的限制
  • NPU算子支持原生算子列表
  • 使用CPU进行图像处理
  • 使用鲲鹏CPU进行视频处理

版本维护策略

维护模式 - 本项目已进入基本维护阶段,不再进行新功能开发,仅进行必要的缺陷修复和安全更新。不再接受新功能提案(RFC)、特性请求或代码贡献。

联系我们

如果有任何疑问或建议,请提交GitCode Issues,我们会尽快回复,感谢您的支持。

安全声明

Torchvision Adapter的系统安全加固、运行用户建议和文件权限控制等内容,请参见Torchvision Adapter 安全声明

免责声明

致Torchvision Adapter插件使用者

  • 本插件仅供调试和开发使用,使用者需自行承担使用风险,并理解以下内容:

    • 数据处理及删除:用户在使用本插件过程中产生的数据属于用户责任范畴。建议用户在使用完毕后及时删除相关数据,以防信息泄露。
    • 数据保密与传播:使用者了解并同意不得将通过本插件产生的数据随意外发或传播。对于由此产生的信息泄露、数据泄露或其他不良后果,本插件及其开发者概不负责。
    • 用户输入安全性:用户需自行保证输入的命令行的安全性,并承担因输入不当而导致的任何安全风险或损失。对于输入命令行不当所导致的问题,本插件及其开发者概不负责。
  • 免责声明范围:本免责声明适用于所有使用本插件的个人或实体。使用本插件即表示您同意并接受本声明的内容,并愿意承担因使用该功能而产生的风险和责任,如有异议请停止使用本插件。

  • 在使用本工具之前,请谨慎阅读并理解以上免责声明的内容。对于使用本插件所产生的任何问题或疑问,请及时联系开发者。

License

Torchvision Adapter插件的使用许可证,具体请参见LICENSE文件。

致谢

感谢来自社区的每一个PR!

项目介绍

本项目开发了Torchvision Adapter插件,用于昇腾适配Torchvision框架。 目前该适配框架增加了对Torchvision所提供的常用算子的支持,提供了基于cv2和基于昇腾NPU的图像处理加速后端以加速图像处理。

定制我的领域