adding extracting images only in cli

pull/132/head
vahidrezanezhad 1 year ago
parent 6aac0b8faf
commit 364ccacab2

@ -67,6 +67,12 @@ from qurator.eynollah.eynollah import Eynollah
is_flag=True,
help="If set, will plot intermediary files and images",
)
@click.option(
"--extract_only_images/--disable-extracting_only_images",
"-eoi/-noeoi",
is_flag=True,
help="If a directory is given, only images in documents will be cropped and saved there and the other processing will not be done",
)
@click.option(
"--allow-enhancement/--no-allow-enhancement",
"-ae/-noae",
@ -148,6 +154,7 @@ def main(
save_layout,
save_deskewed,
save_all,
extract_only_images,
save_page,
enable_plotting,
allow_enhancement,
@ -175,12 +182,16 @@ def main(
if textline_light and not light_version:
print('Error: You used -tll to enable light textline detection but -light is not enabled')
sys.exit(1)
if extract_only_images and not ( save_images and enable_plotting):
print('Error: You used -eoi to enable extract images only mode but did not enable plotting with -ep and providing an output directory with -si')
sys.exit(1)
eynollah = Eynollah(
image_filename=image,
dir_out=out,
dir_in=dir_in,
dir_models=model,
dir_of_cropped_images=save_images,
extract_only_images=extract_only_images,
dir_of_layout=save_layout,
dir_of_deskewed=save_deskewed,
dir_of_all=save_all,

@ -148,6 +148,7 @@ class Eynollah:
dir_out=None,
dir_in=None,
dir_of_cropped_images=None,
extract_only_images=False,
dir_of_layout=None,
dir_of_deskewed=None,
dir_of_all=None,
@ -195,7 +196,7 @@ class Eynollah:
self.allow_scaling = allow_scaling
self.headers_off = headers_off
self.light_version = light_version
self.extract_only_images = True
self.extract_only_images = extract_only_images
self.ignore_page_extraction = ignore_page_extraction
self.pcgts = pcgts
if not dir_in:
@ -2954,9 +2955,6 @@ class Eynollah:
"""
self.logger.debug("enter run")
self.extract_only_images = True
t0_tot = time.time()
if not self.dir_in:

Loading…
Cancel
Save