Update metadata file

README.md

@@ -2,4 +2,10 @@
 license: mit
 ---
 
-Repository: https://github.com/dsgt-kaggle-clef/snakeclef-2024/
+Repository: https://github.com/dsgt-kaggle-clef/snakeclef-2024/
+
+To run the tests:
+
+```bash
+pytest evaluate -vv -x
+```

evaluate/data.py

@@ -36,7 +36,7 @@ class ImageDataset(Dataset):
 
     def __getitem__(self, idx):
         row = self.metadata.iloc[idx]
-        image_path = Path(self.images_root_path) / row.image_path
+        image_path = Path(self.images_root_path) / row.filename
         img = Image.open(image_path).convert("RGB")
         img = v2.ToTensor()(img)
         return {"features": img, "observation_id": row.observation_id}

evaluate/test_evaluate.py

@@ -34,7 +34,7 @@ def images_root(tmp_path):
 def metadata(tmp_path, images_root):
     res = []
     for i, img in enumerate(images_root.glob("*.jpg")):
-        res.append({"image_path": img.name, "observation_id": i})
+        res.append({"filename": img.name, "observation_id": i})
     df = pd.DataFrame(res)
     df.to_csv(tmp_path / "metadata.csv", index=False)
     return tmp_path / "metadata.csv"