|
a |
|
b/Examples/eval_example.py |
|
|
1 |
import sys |
|
|
2 |
sys.path.append('./Radiology_and_AI') |
|
|
3 |
from validation.run_eval import run_eval |
|
|
4 |
import torchio as tio |
|
|
5 |
|
|
|
6 |
validation_transform = tio.Compose([ |
|
|
7 |
tio.ZNormalization(masking_method=tio.ZNormalization.mean), |
|
|
8 |
tio.CropOrPad((240, 240, 160)), |
|
|
9 |
tio.OneHot(num_classes=5) |
|
|
10 |
|
|
|
11 |
]) |
|
|
12 |
|
|
|
13 |
run_eval( |
|
|
14 |
input_data_path= '../brats_new/BraTS2020_TrainingData/MICCAI_BraTS2020_TrainingData', |
|
|
15 |
model_path="../randgamma.pt", |
|
|
16 |
validation_transform=validation_transform, |
|
|
17 |
seed=42, |
|
|
18 |
train_val_split_ration=0.9, |
|
|
19 |
batch_size=1 |
|
|
20 |
) |