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

fixed typo bugs

parent bbf27d78
......@@ -62,9 +62,9 @@ def evaluate_dice_score(trained_model_path,
data_list,
orientation,
prediction_output_path,
device,
LogWriter,
mode):
device= 0,
LogWriter= None,
mode='evaluate'):
"""Dice Score Evaluator
This function evaluates a given trained model by calculating the it's dice score prediction.
......@@ -129,7 +129,7 @@ def evaluate_dice_score(trained_model_path,
if len(volume.shape) == 4:
volume = volume
else:
volume = [:, np.newaxis, :, :]
volume = volume[:, np.newaxis, :, :]
volume = torch.tensor(volume).type(torch.FloatTensor)
label_map = torch.tensor(label_map).type(torhc.LongTensor)
......
......@@ -287,7 +287,7 @@ def preprocess():
As things are - no specific pre-processing currently required!
"""
return pass
return None
def set_orientation(volume, label_map, orientation):
"""Load Data Orientation
......
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