Commit 987c425f authored by Andrei-Claudiu Roibu's avatar Andrei-Claudiu Roibu 🖥
Browse files

function params calls cleanup

parent 018997a3
...@@ -146,6 +146,7 @@ class BrainMapperUNet(nn.Module): ...@@ -146,6 +146,7 @@ class BrainMapperUNet(nn.Module):
Args: Args:
X (torch.tensor): input dMRI volume X (torch.tensor): input dMRI volume
device (int/str): Device type used for training (int - GPU id, str- CPU)
Returns: Returns:
prediction (ndarray): predicted output after training prediction (ndarray): predicted output after training
......
...@@ -191,14 +191,12 @@ def evaluate_dice_score(trained_model_path, ...@@ -191,14 +191,12 @@ def evaluate_dice_score(trained_model_path,
return average_dice_score return average_dice_score
def evaluate_single_path(trained_model_path, def evaluate_single_path(trained_model_path,
number_of_classes,
data_directory, data_directory,
data_list, data_list,
orientation, orientation,
prediction_output_path, prediction_output_path,
batch_size, batch_size,
device= 0, device= 0,
LogWriter= None,
mode='evaluate', mode='evaluate',
exit_on_error = False): exit_on_error = False):
"""Single Path Evaluator """Single Path Evaluator
...@@ -207,14 +205,12 @@ def evaluate_single_path(trained_model_path, ...@@ -207,14 +205,12 @@ def evaluate_single_path(trained_model_path,
Args: Args:
trained_model_path (str): Path to the location of the trained model trained_model_path (str): Path to the location of the trained model
number_of_classes (int): Number of classes
data_directory (str): Path to input data directory data_directory (str): Path to input data directory
data_list (str): Path to a .txt file containing the input files for consideration data_list (str): Path to a .txt file containing the input files for consideration
orientation (str): String detailing the current view (COR, SAG, AXL) orientation (str): String detailing the current view (COR, SAG, AXL)
prediction_output_path (str): Output prediction path prediction_output_path (str): Output prediction path
batch_size (int): Size of batch to be evaluated batch_size (int): Size of batch to be evaluated
device (str/int): Device type used for training (int - GPU id, str- CPU) device (str/int): Device type used for training (int - GPU id, str- CPU)
LogWriter (class): Log Writer class for the BrainMapper U-net
mode (str): Current run mode or phase mode (str): Current run mode or phase
exit_on_error (bool): Flag that triggers the raising of an exception exit_on_error (bool): Flag that triggers the raising of an exception
...@@ -296,7 +292,6 @@ def evaluate_single_path(trained_model_path, ...@@ -296,7 +292,6 @@ def evaluate_single_path(trained_model_path,
def evaluate_two_paths(trained_model1_path, def evaluate_two_paths(trained_model1_path,
trained_model2_path, trained_model2_path,
number_of_classes,
data_directory, data_directory,
data_list, data_list,
orientation1, orientation1,
...@@ -304,7 +299,6 @@ def evaluate_two_paths(trained_model1_path, ...@@ -304,7 +299,6 @@ def evaluate_two_paths(trained_model1_path,
prediction_output_path, prediction_output_path,
batch_size, batch_size,
device= 0, device= 0,
LogWriter= None,
mode='evaluate', mode='evaluate',
exit_on_error = False): exit_on_error = False):
"""Two Path Evaluator """Two Path Evaluator
...@@ -314,7 +308,6 @@ def evaluate_two_paths(trained_model1_path, ...@@ -314,7 +308,6 @@ def evaluate_two_paths(trained_model1_path,
Args: Args:
trained_model1_path (str): Path to the location of the trained model1 trained_model1_path (str): Path to the location of the trained model1
trained_model2_path (str): Path to the location of the trained model2 trained_model2_path (str): Path to the location of the trained model2
number_of_classes (int): Number of classes
data_directory (str): Path to input data directory data_directory (str): Path to input data directory
data_list (str): Path to a .txt file containing the input files for consideration data_list (str): Path to a .txt file containing the input files for consideration
orientation1 (str): String detailing the current view 1 (COR, SAG, AXL) orientation1 (str): String detailing the current view 1 (COR, SAG, AXL)
...@@ -322,7 +315,6 @@ def evaluate_two_paths(trained_model1_path, ...@@ -322,7 +315,6 @@ def evaluate_two_paths(trained_model1_path,
prediction_output_path (str): Output prediction path prediction_output_path (str): Output prediction path
batch_size (int): Size of batch to be evaluated batch_size (int): Size of batch to be evaluated
device (str/int): Device type used for training (int - GPU id, str- CPU) device (str/int): Device type used for training (int - GPU id, str- CPU)
LogWriter (class): Log Writer class for the BrainMapper U-net
mode (str): Current run mode or phase mode (str): Current run mode or phase
exit_on_error (bool): Flag that triggers the raising of an exception exit_on_error (bool): Flag that triggers the raising of an exception
...@@ -419,7 +411,6 @@ def evaluate_two_paths(trained_model1_path, ...@@ -419,7 +411,6 @@ def evaluate_two_paths(trained_model1_path,
def evaluate_all_paths(trained_model1_path, def evaluate_all_paths(trained_model1_path,
trained_model2_path, trained_model2_path,
trained_model3_path, trained_model3_path,
number_of_classes,
data_directory, data_directory,
data_list, data_list,
orientation1, orientation1,
...@@ -428,7 +419,6 @@ def evaluate_all_paths(trained_model1_path, ...@@ -428,7 +419,6 @@ def evaluate_all_paths(trained_model1_path,
prediction_output_path, prediction_output_path,
batch_size, batch_size,
device= 0, device= 0,
LogWriter= None,
mode='evaluate', mode='evaluate',
exit_on_error = False): exit_on_error = False):
"""Two Path Evaluator """Two Path Evaluator
...@@ -439,7 +429,6 @@ def evaluate_all_paths(trained_model1_path, ...@@ -439,7 +429,6 @@ def evaluate_all_paths(trained_model1_path,
trained_model1_path (str): Path to the location of the trained model1 trained_model1_path (str): Path to the location of the trained model1
trained_model2_path (str): Path to the location of the trained model2 trained_model2_path (str): Path to the location of the trained model2
trained_model3_path (str): Path to the location of the trained model3 trained_model3_path (str): Path to the location of the trained model3
number_of_classes (int): Number of classes
data_directory (str): Path to input data directory data_directory (str): Path to input data directory
data_list (str): Path to a .txt file containing the input files for consideration data_list (str): Path to a .txt file containing the input files for consideration
orientation1 (str): String detailing the current view 1 (COR, SAG, AXL) orientation1 (str): String detailing the current view 1 (COR, SAG, AXL)
...@@ -448,7 +437,6 @@ def evaluate_all_paths(trained_model1_path, ...@@ -448,7 +437,6 @@ def evaluate_all_paths(trained_model1_path,
prediction_output_path (str): Output prediction path prediction_output_path (str): Output prediction path
batch_size (int): Size of batch to be evaluated batch_size (int): Size of batch to be evaluated
device (str/int): Device type used for training (int - GPU id, str- CPU) device (str/int): Device type used for training (int - GPU id, str- CPU)
LogWriter (class): Log Writer class for the BrainMapper U-net
mode (str): Current run mode or phase mode (str): Current run mode or phase
exit_on_error (bool): Flag that triggers the raising of an exception exit_on_error (bool): Flag that triggers the raising of an exception
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment