You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
29 lines
870 B
Python
29 lines
870 B
Python
"""
|
|
sbb_binarize CLI
|
|
"""
|
|
|
|
from argparse import ArgumentParser
|
|
|
|
from .sbb_binarize import SbbBinarizer
|
|
|
|
def main():
|
|
parser = ArgumentParser()
|
|
|
|
parser.add_argument('-i', '--image', default=None, help='image.')
|
|
parser.add_argument('-p', '--patches', default=False, help='by setting this parameter to true you let the model to see the image in patches.')
|
|
parser.add_argument('-s', '--save', default=False, help='save prediction with a given name here. The name and format should be given (outputname.tif).')
|
|
parser.add_argument('-m', '--model', default=None, help='models directory.')
|
|
|
|
options = parser.parse_args()
|
|
|
|
binarizer = SbbBinarizer(
|
|
image_path=options.image,
|
|
model=options.model,
|
|
patches=options.patches,
|
|
save=options.save
|
|
)
|
|
binarizer.run()
|
|
|
|
if __name__ == "__main__":
|
|
main()
|