Skip to content

Commit d1338b8

Browse files
committed
Add preprocessing output as model input
1 parent ab20d9b commit d1338b8

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

brats/preprocessing/project/preprocess.py

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
import yaml
66
import numpy as np
77
import nibabel as nib
8+
from shutil import copyfile
89
from tqdm import tqdm
910

1011

@@ -92,9 +93,13 @@ def main():
9293
with open(args.parameters_file, "r") as f:
9394
params = yaml.full_load(f)
9495

95-
images_arr = get_data_arr(args.data_path)
96-
save_processed_data(args.output_path, params["output_filename"], images_arr)
97-
96+
#images_arr = get_data_arr(args.data_path)
97+
#save_processed_data(args.output_path, params["output_filename"], images_arr)
98+
input_file = os.path.normpath(args.data_path + "/BraTS_example_seg.nii.gz")
99+
output_file = os.path.normpath(args.output_path + "/BraTS_example_seg.nii.gz")
100+
print(input_file)
101+
print(output_file)
102+
copyfile(input_file, output_file)
98103

99104
if __name__ == "__main__":
100105
main()

0 commit comments

Comments
 (0)