DDRDATASET

pages/Model_Evaluation.py

@@ -106,6 +106,31 @@ def load_test_data_from_huggingface():
     csv_path = "test_labels_temp.csv"
     df.to_csv(csv_path, index=False)
     dataset = DDRDataset(csv_path=csv_path, transform=val_transform)
+    
+    # ---- DDRDataset Class Definition ----
+    class DDRDataset(Dataset):
+        def __init__(self, csv_path, transform=None):
+            self.data = pd.read_csv(csv_path)
+            self.image_paths = self.data['image_path']
+            self.labels = self.data['label']
+            self.transform = transform
+    
+        def __len__(self):
+            return len(self.data)
+    
+        def __getitem__(self, idx):
+            img_path = self.image_paths[idx]
+            label = self.labels[idx]
+    
+            try:
+                image = Image.open(img_path).convert("RGB")
+            except Exception as e:
+                raise RuntimeError(f"Error loading image from {img_path}: {e}")
+    
+            if self.transform:
+                image = self.transform(image)
+    
+            return image, torch.tensor(label, dtype=torch.long)
     return DataLoader(dataset, batch_size=32, shuffle=False)
 
 # ---- Load Model from Hugging Face (cached) ----