Stefan Wolf
commited on
Commit
·
3d95a88
1
Parent(s):
c16e2e2
Disable score thresholding.
Browse files
script.py
CHANGED
|
@@ -13,7 +13,7 @@ import torch
|
|
| 13 |
# custom script arguments
|
| 14 |
CONFIG_PATH = 'models/swinv2_base_w24_b16x4-fp16_fungi+val_res_384_cb_epochs_6.py'
|
| 15 |
CHECKPOINT_PATH = "models/swinv2_base_w24_b16x4-fp16_fungi+val_res_384_cb_epochs_6_epoch_6_20240514-de00365e.pth"
|
| 16 |
-
SCORE_THRESHOLD = 0.
|
| 17 |
|
| 18 |
def run_inference(input_csv, output_csv, data_root_path):
|
| 19 |
"""Load model and dataloader and run inference."""
|
|
|
|
| 13 |
# custom script arguments
|
| 14 |
CONFIG_PATH = 'models/swinv2_base_w24_b16x4-fp16_fungi+val_res_384_cb_epochs_6.py'
|
| 15 |
CHECKPOINT_PATH = "models/swinv2_base_w24_b16x4-fp16_fungi+val_res_384_cb_epochs_6_epoch_6_20240514-de00365e.pth"
|
| 16 |
+
SCORE_THRESHOLD = 0.0
|
| 17 |
|
| 18 |
def run_inference(input_csv, output_csv, data_root_path):
|
| 19 |
"""Load model and dataloader and run inference."""
|