transforms_use

.gitignore

@@ -1,4 +1,5 @@
 /dataset/
 /dataloader/
 /datasets_tranforms/CIFAR10/
-/datasets_transforms/CIFAR10/
+/datasets_transforms/CIFAR10/
+/datasets_tranforms/p10/

datasets_tranforms/transforms_use_datasets.py

@@ -1,13 +1,22 @@
 import torchvision
-dataset_transform = torchvision.transforms.Compose([torchvision.transforms.ToTensor])
-
-train_set = torchvision.datasets.CIFAR10(root='./CIFAR10',train=True,download=True)
-test_set = torchvision.datasets.CIFAR10(root='./CIFAR10',train=False,download=True)
-
-print(train_set[0])
-print(train_set.classes)
-img,target = train_set[0]
-print(img)
-print(target)
-print(train_set.classes[target])
-img.show()
+from torch.utils.tensorboard import SummaryWriter
+
+dataset_transform = torchvision.transforms.Compose([torchvision.transforms.ToTensor()])
+
+train_set = torchvision.datasets.CIFAR10(root='./CIFAR10',train=True,transform=dataset_transform,download=True)
+test_set = torchvision.datasets.CIFAR10(root='./CIFAR10',train=False,transform=dataset_transform,download=True)
+# print(test_set[0])
+# print(train_set[0])
+# print(train_set.classes)
+# img,target = train_set[0]
+# print(img)
+# print(target)
+# print(train_set.classes[target])
+# img.show()
+
+writer = SummaryWriter('p10')
+for i in range(10):
+    img,target = test_set[i]
+    writer.add_image('test_set',img,i)
+
+writer.close()