fix script
Browse files
script.py
CHANGED
@@ -80,31 +80,31 @@ if __name__ == "__main__":
|
|
80 |
MODEL_NAME = "timm/vit_base_patch14_reg4_dinov2.lvd142m"
|
81 |
|
82 |
# Real submission
|
83 |
-
|
84 |
|
85 |
-
|
86 |
-
|
87 |
-
|
88 |
-
# metadata_file_path = "./test_preprocessed.csv"
|
89 |
-
# test_metadata = pd.read_csv(metadata_file_path)
|
90 |
-
|
91 |
-
# make_submission(
|
92 |
-
# test_metadata=test_metadata,
|
93 |
-
# model_path=MODEL_PATH,
|
94 |
-
# model_name=MODEL_NAME
|
95 |
-
# )
|
96 |
-
|
97 |
-
# Test submission
|
98 |
-
|
99 |
-
metadata_file_path = "../trial_submission.csv"
|
100 |
|
|
|
101 |
test_metadata = pd.read_csv(metadata_file_path)
|
102 |
|
103 |
make_submission(
|
104 |
test_metadata=test_metadata,
|
105 |
model_path=MODEL_PATH,
|
106 |
-
model_name=MODEL_NAME
|
107 |
-
images_root_path="../data/DF_FULL"
|
108 |
)
|
109 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
110 |
|
|
|
80 |
MODEL_NAME = "timm/vit_base_patch14_reg4_dinov2.lvd142m"
|
81 |
|
82 |
# Real submission
|
83 |
+
import zipfile
|
84 |
|
85 |
+
with zipfile.ZipFile("/tmp/data/private_testset.zip", 'r') as zip_ref:
|
86 |
+
zip_ref.extractall("/tmp/data")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
87 |
|
88 |
+
metadata_file_path = "./_test_preprocessed.csv"
|
89 |
test_metadata = pd.read_csv(metadata_file_path)
|
90 |
|
91 |
make_submission(
|
92 |
test_metadata=test_metadata,
|
93 |
model_path=MODEL_PATH,
|
94 |
+
model_name=MODEL_NAME
|
|
|
95 |
)
|
96 |
|
97 |
+
# # Test submission
|
98 |
+
|
99 |
+
# metadata_file_path = "../trial_submission.csv"
|
100 |
+
|
101 |
+
# test_metadata = pd.read_csv(metadata_file_path)
|
102 |
+
|
103 |
+
# make_submission(
|
104 |
+
# test_metadata=test_metadata,
|
105 |
+
# model_path=MODEL_PATH,
|
106 |
+
# model_name=MODEL_NAME,
|
107 |
+
# images_root_path="../data/DF_FULL"
|
108 |
+
# )
|
109 |
+
|
110 |
|