imp:unziping dataset

script.py

@@ -1,5 +1,3 @@
-import glob
-import traceback
 import pandas as pd
 import numpy as np
 from PIL import Image
@@ -37,7 +35,7 @@ class ONNXWorker:
         return logits.tolist()
 
 
-def make_submission(test_metadata, model_path, output_csv_path="./submission.csv", data_root_path="/tmp/data"):
+def make_submission(test_metadata, model_path, output_csv_path="./submission.csv", images_root_path="/tmp/data/private_testset"):
     """Make submission with given """
 
     model = ONNXWorker(model_path)
@@ -45,7 +43,7 @@ def make_submission(test_metadata, model_path, output_csv_path="./submission.csv
     predictions = []
 
     for _, row in tqdm(test_metadata.iterrows(), total=len(test_metadata)):
-        image_path = os.path.join(data_root_path, row.filename)
+        image_path = os.path.join(images_root_path, row.filename)
         test_image = np.asarray(Image.open(image_path).convert("RGB"))
         logits = model.predict_image(test_image)
 
@@ -59,6 +57,11 @@ def make_submission(test_metadata, model_path, output_csv_path="./submission.csv
 
 if __name__ == "__main__":
 
+    import zipfile
+
+    with zipfile.ZipFile("/tmp/data/private_testset.zip", 'r') as zip_ref:
+        zip_ref.extractall("/tmp/data")
+
     ONNX_MODEL_PATH = "./swinv2_tiny_window16_256.onnx"
 
     metadata_file_path = "./SnakeCLEF2024-TestMetadata.csv"