Pytorch transform

Pytorch transform.txt

from torchvision import transforms

transform = transforms.Compose([
    transforms.ToTensor(),  # Bu metot PIL.Image veya numpy.ndarray (H x W x C) [0, 255] verilerini
                             # torch.FloatTensor (C x H x W) [0.0, 1.0] formatına dönüştürür.
])

class CustomDataset(torch.utils.data.Dataset):
    def __init__(self, image_paths, transform=None):
        self.image_paths = image_paths
        self.transform = transform

    def __len__(self):
        return len(self.image_paths)

    def __getitem__(self, index):
        image_path = self.image_paths[index]
        image = Image.open(image_path)  # PIL.Image olarak aç
        if self.transform:
            image = self.transform(image)  # transform uygula, örneğin ToTensor()
        
        return image

# Kullanım örneği
dataset = CustomDataset(my_image_paths, transform=transform)
dataloader = DataLoader(dataset, batch_size=10, shuffle=True)