delet ckpt

metaformer-s-224.ckpt

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:30285f565bdeb54f1ce8bfea244bca3d47c04746a649cd319d02f31ec553fd49
-size 331278138

script.py

@@ -15,11 +15,13 @@ def is_gpu_available():
 WIDTH = 224
 HEIGHT = 224
 
+MODEL_PATH = "metaformer-s-224.pth"
+MODEL_NAME = "caformer_s18.sail_in22k"
 
 class PytorchWorker:
     """Run inference using ONNX runtime."""
 
-    def __init__(self, model_path: str, model_name: str, number_of_categories: int = 1604):
+    def __init__(self, model_path: str, model_name: str, number_of_categories: int = 1605):
 
         def _load_model(model_name, model_path):
 
@@ -60,7 +62,7 @@ def make_submission(test_metadata, model_path, model_name, output_csv_path="./su
     predictions = []
 
     for _, row in tqdm(test_metadata.iterrows(), total=len(test_metadata)):
-        image_path = os.path.join(images_root_path, row.image_path)
+        image_path = os.path.join(images_root_path, row.image_path.replace("jpg", "JPG"))
 
         test_image = Image.open(image_path).convert("RGB")
 
@@ -73,17 +75,28 @@ def make_submission(test_metadata, model_path, model_name, output_csv_path="./su
     user_pred_df = test_metadata.drop_duplicates("observation_id", keep="first")
     user_pred_df[["observation_id", "class_id"]].to_csv(output_csv_path, index=None)
 
+def test_submission():
+
+    metadata_file_path = "../val_mini.csv"
+    test_metadata = pd.read_csv(metadata_file_path)
+
+    make_submission(
+        test_metadata=test_metadata,
+        model_path=MODEL_PATH,
+        model_name=MODEL_NAME,
+        images_root_path="../data/DF_FULL/"
+    )
+
 
 if __name__ == "__main__":
 
+    # test_submission()
+
     import zipfile
 
     with zipfile.ZipFile("/tmp/data/private_testset.zip", 'r') as zip_ref:
         zip_ref.extractall("/tmp/data")
 
-    MODEL_PATH = "metaformer-s-224.pth"
-    MODEL_NAME = "caformer_s18.sail_in22k"
-
     metadata_file_path = "./FungiCLEF2024_TestMetadata.csv"
     test_metadata = pd.read_csv(metadata_file_path)