picekl commited on
Commit
3cf8c5c
1 Parent(s): 9f0d310

feat: final working version

Browse files
Files changed (1) hide show
  1. script.py +2 -2
script.py CHANGED
@@ -38,7 +38,7 @@ class PytorchWorker:
38
 
39
  self.model = _load_model(model_name, model_path)
40
 
41
- self.transforms = T.Compose([T.Resize((224, 224)),
42
  T.ToTensor(),
43
  T.Normalize([0.5, 0.5, 0.5], [0.5, 0.5, 0.5])])
44
 
@@ -88,7 +88,7 @@ if __name__ == "__main__":
88
  MODEL_NAME = "tf_efficientnet_b1.ap_in1k"
89
 
90
  metadata_file_path = "./FungiCLEF2024_TestMetadata.csv"
91
- test_metadata = pd.read_csv(metadata_file_path)[:100]
92
 
93
  make_submission(
94
  test_metadata=test_metadata,
 
38
 
39
  self.model = _load_model(model_name, model_path)
40
 
41
+ self.transforms = T.Compose([T.Resize((299, 299)),
42
  T.ToTensor(),
43
  T.Normalize([0.5, 0.5, 0.5], [0.5, 0.5, 0.5])])
44
 
 
88
  MODEL_NAME = "tf_efficientnet_b1.ap_in1k"
89
 
90
  metadata_file_path = "./FungiCLEF2024_TestMetadata.csv"
91
+ test_metadata = pd.read_csv(metadata_file_path)
92
 
93
  make_submission(
94
  test_metadata=test_metadata,