chychiu commited on
Commit
57e0d82
·
1 Parent(s): 3735d66

fix script

Browse files
Files changed (1) hide show
  1. script.py +18 -18
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
- # 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
 
 
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