From c6989dbeeb3f21d745223af5955816b71d92d20c Mon Sep 17 00:00:00 2001 From: holy2020 Date: Wed, 8 Jun 2022 00:43:57 +0200 Subject: [PATCH] test-decker.py --- test_docker.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/test_docker.py b/test_docker.py index 407c5d1..674a00f 100644 --- a/test_docker.py +++ b/test_docker.py @@ -5,11 +5,13 @@ import os import sys +import warnings +warnings.filterwarnings('ignore') from src.LFBNet.utilities import train_valid_paths from src.LFBNet.preprocessing import preprocessing from src.run import trainer, parse_argument -from src.LFBNet.utilities.compute_surrogate_features import ComputesTMTVsDmaxFromnii +from src.LFBNet.utilities.compute_surrogate_features import ComputesTMTVsDmaxFromNii def main(): @@ -52,7 +54,7 @@ def main(): """ # Path to the parent/main directory. Please read readme.md for how to organize your files. - input_dir = "/input_data" + input_dir = "/input" # parameters to set dataset_name = 'data' @@ -80,11 +82,11 @@ def main(): print("\n\n Computing the surrogate biomarkers ... \n\n") for identifier, data_path in zip( - ["prd", "gt"], [os.path.join(preprocessing_data_dir, "predicted_data"), + ["predicted", "ground_truth"], [os.path.join(preprocessing_data_dir, "predicted_data"), os.path.join(preprocessing_data_dir, "data_default_MIP_dir")] ): try: - csv_file = ComputesTMTVsDmaxFromnii(data_path=data_path, get_identifier=identifier) + csv_file = ComputesTMTVsDmaxFromNii(data_path=data_path, get_identifier=identifier) csv_file.compute_and_save_surrogate_features() except: continue