mirror of
https://github.com/qurator-spk/sbb_binarization.git
synced 2025-07-04 00:19:53 +02:00
24 lines
786 B
Python
24 lines
786 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(options.image, options.model, options.patches, options.save)
|
||
|
|
binarizer.run()
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|