include transform

Author: Dnafivuq

task_3/taskdata.py

@@ -1,6 +1,7 @@
 from typing import Tuple
 import torch
 from torch.utils.data import Dataset
+from torchvision import transforms
 
 
 class TaskDataset(Dataset):
@@ -21,4 +22,12 @@ def __getitem__(self, index) -> Tuple[int, torch.Tensor, int]:
         return id_, img, label
 
     def __len__(self):
-        return len(self.ids)
+        return len(self.ids)
+    
+t = transforms.Compose(
+    [
+        transforms.Resize((32, 32)),
+        transforms.Lambda(lambda x: x.convert("RGB")),
+        transforms.ToTensor(),
+    ]
+)