Spaces:
Sleeping
Sleeping
File size: 1,082 Bytes
4719ce5 9f284dc 9921695 9f284dc d949fe4 7245c5a 4719ce5 7245c5a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
import sys
import argparse
import os
from lungtumormask import mask
def path(string):
if os.path.exists(string):
return string
else:
sys.exit(f'File not found: {string}')
def main():
parser = argparse.ArgumentParser()
parser.add_argument('input', metavar='input', type=path, help='Path to the input image, should be .nifti')
parser.add_argument('output', metavar='output', type=str, help='Filepath for output tumormask')
parser.add_argument('--lung-filter', action='store_true', help='whether to apply lungmask postprocessing.')
parser.add_argument('--threshold', metavar='threshold', type=float, default=0.5,
help='which threshold to use for assigning voxel-wise classes.')
parser.add_argument('--radius', metavar='radius', type=int, default=1,
help='which radius to use for morphological post-processing segmentation smoothing.')
argsin = sys.argv[1:]
args = parser.parse_args(argsin)
mask.mask(args.input, args.output, args.lung_filter, args.threshold, args.radius)
|