質問編集履歴
1
コードの修正
test
CHANGED
File without changes
|
test
CHANGED
@@ -72,6 +72,38 @@
|
|
72
72
|
|
73
73
|
|
74
74
|
|
75
|
+
class DownSizePairImageFolder(ImageFolder):
|
76
|
+
|
77
|
+
def __init__(self, root, transform=None, large_size=256, small_size=64, **kwds):
|
78
|
+
|
79
|
+
super().__init__(root, transform=transform, **kwds)
|
80
|
+
|
81
|
+
self.large_resizer = transforms.Scale(large_size)
|
82
|
+
|
83
|
+
self.small_resizer = transforms.Scale(small_size)
|
84
|
+
|
85
|
+
|
86
|
+
|
87
|
+
def __getitem__(self, index):
|
88
|
+
|
89
|
+
path, _ = self.imgs[index]
|
90
|
+
|
91
|
+
img = self.loader(path)
|
92
|
+
|
93
|
+
large_img = self.large_resizer(img)
|
94
|
+
|
95
|
+
small_img = self.small_resizer(img)
|
96
|
+
|
97
|
+
if self.transform is not None:
|
98
|
+
|
99
|
+
large_img = self.transform(large_img)
|
100
|
+
|
101
|
+
small_img = self.transform(small_img)
|
102
|
+
|
103
|
+
return small_img, large_img
|
104
|
+
|
105
|
+
|
106
|
+
|
75
107
|
train_data = DownSizePairImageFolder('C:/SRGAN/train/image', transform=transforms.ToTensor())
|
76
108
|
|
77
109
|
test_data = DownSizePairImageFolder('C:/SRGAN/train/image', transform=transforms.ToTensor())
|