Torchvision veri kümeleri, makine öğrenimi modellerini geliştirmek ve test etmek için bilgisayarlı görmede yaygın olarak kullanılan popüler veri kümelerinin koleksiyonlarıdır. Geliştiriciler torchvision veri kümeleriyle makine öğrenimi modellerini görüntü sınıflandırma, nesne algılama ve segmentasyon gibi çeşitli görevlerde eğitebilir ve test edebilir.
Bu veri kümesine erişmek için doğrudan şu adresten indirebilirsiniz:
import torchvision.datasets as datasets # Load the training dataset train_dataset = datasets.MNIST(root='data/', train=True, transform=None, download=True) # Load the testing dataset test_dataset = datasets.MNIST(root='data/', train=False, transform=None, download=True)
PyTorch torchvision paketini kullanarak MNIST veri kümesini yüklemek için kod. 20/3/2023 tarihinde adresinden alındı.
Bu veri kümesi şu adresten indirilebilir:
import torch import torchvision import torchvision.transforms as transforms transform = transforms.Compose( [transforms.ToTensor(), transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))]) trainset = torchvision.datasets.CIFAR10(root='./data', train=True, download=True, transform=transform) testset = torchvision.datasets.CIFAR10(root='./data', train=False, download=True, transform=transform) trainloader = torch.utils.data.DataLoader(trainset, batch_size=4, shuffle=True, num_workers=2) testloader = torch.utils.data.DataLoader(testset, batch_size=4, shuffle=False, num_workers=2)
PyTorch torchvision paketini kullanarak CIFAR-10 veri kümesini yüklemek için kod. adresinden 20/3/2023 tarihinde alındı.
Torchvision veri kümesini Kaggle'dan indirmek için lütfen Kaggle'ı ziyaret edin.
import torchvision.datasets as datasets import torchvision.transforms as transforms # Define transform to normalize data transform = transforms.Compose([ transforms.ToTensor(), transforms.Normalize(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5]) ]) # Load CIFAR-100 train and test datasets trainset = datasets.CIFAR100(root='./data', train=True, download=True, transform=transform) testset = datasets.CIFAR100(root='./data', train=False, download=True, transform=transform) # Create data loaders for train and test datasets trainloader = torch.utils.data.DataLoader(trainset, batch_size=64, shuffle=True) testloader = torch.utils.data.DataLoader(testset, batch_size=64, shuffle=False)
PyTorch torchvision paketini kullanarak CIFAR-100 veri kümesini yüklemek için kod. 20/3/2023 tarihinde adresinden alındı.
Bu torchvision veri kümesini indirmek için şu adresi ziyaret etmeniz gerekir:
import torchvision.datasets as datasets import torchvision.transforms as transforms # Set the path to the ImageNet dataset on your machine data_path = "/path/to/imagenet" # Create the ImageNet dataset object with custom options imagenet_train = datasets.ImageNet( root=data_path, split='train', transform=transforms.Compose([ transforms.Resize(256), transforms.RandomCrop(224), transforms.RandomHorizontalFlip(), transforms.ToTensor(), transforms.Normalize( mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]) ]), download=False ) imagenet_val = datasets.ImageNet( root=data_path, split='val', transform=transforms.Compose([ transforms.Resize(256), transforms.CenterCrop(224), transforms.ToTensor(), transforms.Normalize( mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]) ]), download=False ) # Print the number of images in the training and validation sets print("Number of images in the training set:", len(imagenet_train)) print("Number of images in the validation set:", len(imagenet_val))
PyTorch torchvision paketini kullanarak ImageNet veri kümesini yüklemek için kod. adresinden 21/3/2023 tarihinde alındı.
Bu torchvision veri kümesini indirmek için lütfen şu adresi ziyaret edin:
import torch from torchvision import datasets, transforms # Define transformation transform = transforms.Compose([ transforms.Resize(256), transforms.CenterCrop(224), transforms.ToTensor(), transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]) ]) # Load training dataset train_dataset = datasets.CocoDetection(root='/path/to/dataset/train2017', annFile='/path/to/dataset/annotations/instances_train2017.json', transform=transform) train_loader = torch.utils.data.DataLoader(train_dataset, batch_size=32, shuffle=True) # Load validation dataset val_dataset = datasets.CocoDetection(root='/path/to/dataset/val2017', annFile='/path/to/dataset/annotations/instances_val2017.json', transform=transform) val_loader = torch.utils.data.DataLoader(val_dataset, batch_size=32, shuffle=False)
PyTorch torchvision paketini kullanarak MS Coco veri kümesini yüklemek için kod. 21/3/2023 tarihinde adresinden alındı.
Bu torchvision veri kümesi şu adresten indirilebilir:
import torch import torchvision import torchvision.transforms as transforms # Define transformations transform = transforms.Compose( [transforms.ToTensor(), transforms.Normalize((0.5,), (0.5,))]) # Load the dataset trainset = torchvision.datasets.FashionMNIST(root='./data', train=True, download=True, transform=transform) testset = torchvision.datasets.FashionMNIST(root='./data', train=False, download=True, transform=transform) # Create data loaders trainloader = torch.utils.data.DataLoader(trainset, batch_size=4, shuffle=True, num_workers=2) testloader = torch.utils.data.DataLoader(testset, batch_size=4, shuffle=False, num_workers=2)
Fashion-MNIST veri kümesini PyTorch torchvision paketini kullanarak yüklemeye yönelik kod. 21/3/2023 tarihinde adresinden alındı.
Bu torchvision veri kümesini indirmek için şu adrese gidebilirsiniz:
import torchvision import torch # Load the train and test sets train_set = torchvision.datasets.SVHN(root='./data', split='train', download=True, transform=torchvision.transforms.ToTensor()) test_set = torchvision.datasets.SVHN(root='./data', split='test', download=True, transform=torchvision.transforms.ToTensor()) # Create data loaders train_loader = torch.utils.data.DataLoader(train_set, batch_size=64, shuffle=True) test_loader = torch.utils.data.DataLoader(test_set, batch_size=64, shuffle=False)
PyTorch torchvision paketini kullanarak SVHN veri kümesini yüklemek için kod. 22/3/2023 tarihinde adresinden alındı.
Bu veri kümesine erişmek için doğrudan şu adresten indirebilirsiniz:
import torchvision.datasets as datasets import torchvision.transforms as transforms # Define the transformation to apply to the data transform = transforms.Compose([ transforms.ToTensor(), # Convert PIL image to PyTorch tensor transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)) # Normalize the data ]) # Load the STL-10 dataset train_dataset = datasets.STL10(root='./data', split='train', download=True, transform=transform) test_dataset = datasets.STL10(root='./data', split='test', download=True, transform=transform)
PyTorch torchvision paketini kullanarak STL-10 veri kümesini yüklemek için kod. adresinden 22/3/2023 tarihinde alındı.
Bu veri kümesini şu adresten indirebilirsiniz:
import torchvision.datasets as datasets import torchvision.transforms as transforms transform = transforms.Compose([ transforms.CenterCrop(178), transforms.Resize(128), transforms.ToTensor(), transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)) ]) celeba_dataset = datasets.CelebA(root='./data', split='train', transform=transform, download=True)
PyTorch torchvision paketini kullanarak CelebA veri kümesini yüklemek için kod. adresinden 22/3/2023 tarihinde alındı.
Güncel veri kümesine erişmek için şu adresten indirebilirsiniz:
import torch import torchvision from torchvision import transforms # Define transformations to apply to the images transform = transforms.Compose([ transforms.Resize((224, 224)), transforms.ToTensor(), transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]) ]) # Load the train and validation datasets train_dataset = torchvision.datasets.VOCDetection(root='./data', year='2007', image_set='train', transform=transform) val_dataset = torchvision.datasets.VOCDetection(root='./data', year='2007', image_set='val', transform=transform) # Create data loaders train_loader = torch.utils.data.DataLoader(train_dataset, batch_size=32, shuffle=True) val_loader = torch.utils.data.DataLoader(val_dataset, batch_size=32, shuffle=False)
PyTorch torchvision paketini kullanarak PASCAL VOC veri kümesini yüklemek için kod. 22/3/2023 tarihinde adresinden alındı.
Bu veri kümesine erişmek için şunları kullanabilirsiniz:
import torch import torchvision from torchvision import transforms # Define transformations to apply to the images transform = transforms.Compose([ transforms.Resize((224, 224)), transforms.ToTensor(), transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]) ]) # Load the train and validation datasets train_dataset = torchvision.datasets.Places365(root='./data', split='train-standard', transform=transform) val_dataset = torchvision.datasets.Places365(root='./data', split='val', transform=transform) # Create data loaders train_loader = torch.utils.data.DataLoader(train_dataset, batch_size=32, shuffle=True) val_loader = torch.utils.data.DataLoader(val_dataset, batch_size=32, shuffle=False)
PyTorch torchvision paketini kullanarak Places365 veri kümesini yükleme kodu. adresinden 22/3/2023 tarihinde alındı.
Bu makalenin ana görseli , HackerNoon'un AI Stabil Difüzyon modeli aracılığıyla 'küçük kareler halinde bir araya getirilen binlerce görüntü' istemi kullanılarak oluşturuldu.
Daha Fazla Veri Kümesi Listesi: