torchvision (ecosystem)

E96634

torchvision is a PyTorch-based computer vision library providing datasets, model architectures, and image transformations commonly used for training and evaluating deep learning models.

All labels observed (7)

How this entity was disambiguated

Statements (74)

Predicate Object
instanceOf PyTorch ecosystem component
computer vision library
software library
dataset CIFAR10
linked to: CIFAR-10

CIFAR100
linked to: CIFAR-100

COCO
CelebA
FashionMNIST
linked to: Fashion-MNIST

ImageNet
MNIST
SVHN
VOCDetection
VOCSegmentation
designedFor deep learning researchers
machine learning practitioners
domain computer vision
framework PyTorch
hasSubmodule torchvision.datasets
torchvision.io
torchvision.models
torchvision.ops
torchvision.transforms
implements image classification models
image transformations
instance segmentation models
keypoint detection models
object detection models
semantic segmentation models
video transformations
integratesWith torch.utils.data.DataLoader
license BSD-style license
modelFamily DenseNet
FasterRCNN
GoogLeNet
InceptionV3
KeypointRCNN
MaskRCNN
MobileNetV2
ResNet
RetinaNet
ShuffleNetV2
SqueezeNet
VGG
partOf PyTorch ecosystem
linked to: PyTorch
programmingLanguage Python
provides data loading utilities
model evaluation utilities
pretrained models
standard vision datasets
repositoryPlatform GitHub
supports CPU
CUDA
linked to: NVIDIA CUDA

GPU
TorchScript
linked to: PyTorch
supportsTask image classification
instance segmentation
keypoint detection
object detection
semantic segmentation
video classification
transformType CenterCrop
ColorJitter
Compose
Normalize
RandomCrop
RandomHorizontalFlip
RandomResizedCrop
RandomRotation
Resize
ToTensor
useCase benchmarking on standard datasets
data augmentation for images
evaluating computer vision models
training deep learning vision models

How these facts were elicited

Referenced by (7)

Full triples — surface form annotated when it differs from this entity's canonical label.

PyTorch hasComponent torchvision (ecosystem)
torchvision hasSubmodule torchvision.datasets
subject linked to: torchvision (ecosystem)
linked to: torchvision (ecosystem)
torchvision hasSubmodule torchvision.models
subject linked to: torchvision (ecosystem)
linked to: torchvision (ecosystem)
torchvision hasSubmodule torchvision.transforms
subject linked to: torchvision (ecosystem)
linked to: torchvision (ecosystem)
torchvision hasSubmodule torchvision.io
subject linked to: torchvision (ecosystem)
linked to: torchvision (ecosystem)
torchvision hasSubmodule torchvision.ops
subject linked to: torchvision (ecosystem)
linked to: torchvision (ecosystem)
CIFAR-100 includedIn TorchVision datasets
linked to: torchvision (ecosystem)