Skip to content

Commit

Permalink
add parameter to use local model in debug script test_models.py
Browse files Browse the repository at this point in the history
  • Loading branch information
2320sharon committed Dec 11, 2024
1 parent 88defdc commit 5a396b3
Showing 1 changed file with 9 additions and 2 deletions.
11 changes: 9 additions & 2 deletions debug_scripts/test_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -83,6 +83,7 @@ def test_model(img_types:list, available_models_dict:dict, input_directory:str):
"model_type": model_selected,
"otsu": False,
"tta": False,
"use_local_model": False,
}

zoo_model_instance = zoo_model.Zoo_Model()
Expand All @@ -106,13 +107,19 @@ def test_model(img_types:list, available_models_dict:dict, input_directory:str):
"AK_segformer_RGB_4class_14037041", # AK segformer model
],
"MNDWI": [
"segformer_MNDWI_4class_8213443",
"global_segformer_MNDWI_4class_14183366", # global segformer model
"AK_segformer_MNDWI_4class_14187478", # AK segformer model
],
"NDWI": [
"segformer_NDWI_4class_8213427",
"global_segformer_NDWI_4class_14172182", # global segformer model
"AK_segformer_NDWI_4class_14183210", # AK segformer model
],
}



img_types = ["RGB", "MNDWI", "NDWI"]
# img_types = [ "NDWI"]
# img_types = [ "MNDWI"]

test_model(img_types, available_models_dict, input_directory)

0 comments on commit 5a396b3

Please sign in to comment.