Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -69,6 +69,17 @@ class Pix2PixDataset(torch.utils.data.Dataset):
|
|
69 |
assert len(self.originals) == len(self.targets)
|
70 |
print(f"Number of original images: {len(self.originals)}")
|
71 |
print(f"Number of target images: {len(self.targets)}")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
72 |
self.transform = transform
|
73 |
|
74 |
def __len__(self):
|
|
|
69 |
assert len(self.originals) == len(self.targets)
|
70 |
print(f"Number of original images: {len(self.originals)}")
|
71 |
print(f"Number of target images: {len(self.targets)}")
|
72 |
+
|
73 |
+
# Debugging: Print out filenames from the dataset and CSV
|
74 |
+
print("Dataset Original Filenames:")
|
75 |
+
for original in self.originals:
|
76 |
+
print(original['image'].filename)
|
77 |
+
|
78 |
+
print("\nCSV Image Filenames:")
|
79 |
+
print(self.data['image_path'].unique())
|
80 |
+
|
81 |
+
|
82 |
+
|
83 |
self.transform = transform
|
84 |
|
85 |
def __len__(self):
|