qurator-spk/sbb_binarization
1
0
Fork 0
mirror of https://github.com/qurator-spk/sbb_binarization.git synced 2025-06-09 04:10:00 +02:00
Code Issues Releases Wiki Activity

put CLI into its own module

Browse source
This commit is contained in:
Konstantin Baierer 2020-10-15 15:38:49 +02:00
parent 71d44408b3
commit 389ef088d0
3 changed files with 25 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

23
sbb_binarize/cli.py Normal file
View file

@ -0,0 +1,23 @@
"""
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()

21
sbb_binarize/sbb_binarize.py
View file

@ -2,7 +2,6 @@
Tool to load model and binarize a given image.
"""
from argparse import ArgumentParser
from os import listdir
from os.path import join
from warnings import catch_warnings, simplefilter
@ -16,7 +15,7 @@ import tensorflow as tf
with catch_warnings():
simplefilter("ignore")
class sbb_binarize:
class SbbBinarizer:
# TODO use True/False for patches
def __init__(self, image, model, patches='false', save=None):
@ -218,21 +217,3 @@ class sbb_binarize:
img_last = (img_last[:, :] == 0)*255
if self.save:
cv2.imwrite(self.save, img_last)
def main():
parser = ArgumentParser()
parser.add_argument('-i', '--image', dest='inp1', default=None, help='image.')
parser.add_argument('-p', '--patches', dest='inp3', default=False, help='by setting this parameter to true you let the model to see the image in patches.')
parser.add_argument('-s', '--save', dest='inp4', default=False, help='save prediction with a given name here. The name and format should be given (outputname.tif).')
parser.add_argument('-m', '--model', dest='inp2', default=None, help='models directory.')
options = parser.parse_args()
possibles = globals()
possibles.update(locals())
x = sbb_binarize(options.inp1, options.inp2, options.inp3, options.inp4)
x.run()
if __name__ == "__main__":
main()