Spaces:
Running
Running
Vemund Fredriksen
commited on
Commit
•
0d8deb6
1
Parent(s):
c35dad1
Remove main method
Browse files- lungtumormask/dataprocessing.py +1 -16
- lungtumormask/mask.py +1 -4
lungtumormask/dataprocessing.py
CHANGED
@@ -247,19 +247,4 @@ def post_process(left_mask, right_mask, preprocess_dump):
|
|
247 |
|
248 |
stitched = np.logical_or(left, right).astype(int)
|
249 |
|
250 |
-
return stitched
|
251 |
-
|
252 |
-
|
253 |
-
if __name__ == "__main__":
|
254 |
-
path = "D:\\Datasets\MSD\\Images\\lung_003.nii.gz"
|
255 |
-
preprocess_dump = preprocess(path)
|
256 |
-
|
257 |
-
unpad = post_process(preprocess_dump['left_lung'], preprocess_dump['right_lung'], preprocess_dump)
|
258 |
-
|
259 |
-
import nibabel
|
260 |
-
|
261 |
-
nimage = nibabel.Nifti1Image(unpad, preprocess_dump['org_affine'])
|
262 |
-
nibabel.save(nimage, "D:\\Datasets\\stitched.nii.gz")
|
263 |
-
|
264 |
-
|
265 |
-
|
|
|
247 |
|
248 |
stitched = np.logical_or(left, right).astype(int)
|
249 |
|
250 |
+
return stitched
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
lungtumormask/mask.py
CHANGED
@@ -28,7 +28,4 @@ def mask(image_path, save_path):
|
|
28 |
|
29 |
print(f"Storing segmentation at {save_path}")
|
30 |
nimage = nibabel.Nifti1Image(infered, preprocess_dump['org_affine'])
|
31 |
-
nibabel.save(nimage, save_path)
|
32 |
-
|
33 |
-
if __name__ == "__main__":
|
34 |
-
mask("D:\\Datasets\\MSD\\Images\\lung_003.nii.gz", "D:\\Datasets\\MSD\\Images\\out3.nii.gz")
|
|
|
28 |
|
29 |
print(f"Storing segmentation at {save_path}")
|
30 |
nimage = nibabel.Nifti1Image(infered, preprocess_dump['org_affine'])
|
31 |
+
nibabel.save(nimage, save_path)
|
|
|
|
|
|