3v324v23 commited on
Commit
09a8a20
·
1 Parent(s): b925d8f

add image path to test

Browse files
Files changed (1) hide show
  1. pages/Model_Evaluation.py +5 -0
pages/Model_Evaluation.py CHANGED
@@ -102,8 +102,13 @@ def load_test_data_from_huggingface():
102
  split="test"
103
  )
104
  df = dataset.to_pandas()
 
 
 
 
105
  csv_path = "test_labels_temp.csv"
106
  df.to_csv(csv_path, index=False)
 
107
  dataset = DDRDataset(csv_path=csv_path, transform=val_transform)
108
  return DataLoader(dataset, batch_size=32, shuffle=False)
109
 
 
102
  split="test"
103
  )
104
  df = dataset.to_pandas()
105
+
106
+ # 🔧 Add correct relative path to image files
107
+ df["new_path"] = "test/" + df["image"].astype(str)
108
+
109
  csv_path = "test_labels_temp.csv"
110
  df.to_csv(csv_path, index=False)
111
+
112
  dataset = DDRDataset(csv_path=csv_path, transform=val_transform)
113
  return DataLoader(dataset, batch_size=32, shuffle=False)
114