pytorch

vision

pytorch

Datasets, Transforms and Models specific to Computer Vision

AI 简介

torchvision 是一个专注于计算机视觉任务的 PyTorch 库,提供了常用的数据集、模型架构以及图像变换工具。其核心功能包括对多种流行数据集的支持(如 CIFAR-10, ImageNet 等),预定义的经典及最新的深度学习模型(如 ResNet, VGG, Faster R-CNN 等),还有丰富的图像处理操作,这些都极大地简化了从数据准备到模型训练再到结果分析的过程。此外,torchvision 支持使用 Torch 张量和 PIL 图像作为后端,确保了灵活性与效率。此库非常适合用于学术研究、教育目的或是工业界中需要快速原型设计和部署计算机视觉应用的场景。

Python
BSD 3-Clause "New" or "Revised" License
17.7k
Stars
7.2k
Forks
439
Watchers
882
Issues

Star 增长

今日+7
近 7 天+21
近 30 天+70
综合评分101.5
默认分支main