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

🚧 sbb_textline_detector: Get image via image_from_page

Browse source
This commit is contained in:
Gerber, Mike 2020-10-15 18:57:21 +02:00
parent 38df5dbbfd
commit a9b9c8a885
1 changed files with 12 additions and 10 deletions

22
qurator/sbb_textline_detector/ocrd_cli.py
View file

@ -38,14 +38,6 @@ class OcrdSbbTextlineDetectorRecognize(Processor):
kwargs['version'] = OCRD_TOOL['version']
super(OcrdSbbTextlineDetectorRecognize, self).__init__(*args, **kwargs)
def _resolve_image_file(self, input_file: OcrdFile) -> str:
if input_file.mimetype == MIMETYPE_PAGE:
pcgts = page_from_file(self.workspace.download_file(input_file))
page = pcgts.get_Page()
image_file = page.imageFilename
else:
image_file = input_file.local_filename
return image_file
def process(self):
log = getLogger('processor.OcrdSbbTextlineDetectorRecognize')
@ -64,9 +56,20 @@ class OcrdSbbTextlineDetectorRecognize(Processor):
except FileExistsError:
pass
pcgts = page_from_file(self.workspace.download_file(input_file))
page = pcgts.get_Page()
page_image, page_coords, page_image_info = \
self.workspace.image_from_page(
page, page_id,
feature_filter='binarized,grayscale_normalized'
)
with tempfile.TemporaryDirectory() as tmp_dirname:
# Save the image
image_file = tempfile.mkstemp(dir=tmp_dirname, suffix='.png')[1]
page_image.save(image_file)
# Segment the image
image_file = self._resolve_image_file(input_file)
model = self.parameter['model']
x = textline_detector(image_file, tmp_dirname, file_id, model)
x.run()
@ -77,7 +80,6 @@ class OcrdSbbTextlineDetectorRecognize(Processor):
tmp_page = tmp_pcgts.get_Page()
# Create a new PAGE file from the input file
pcgts = page_from_file(self.workspace.download_file(input_file))
pcgts.set_pcGtsId(file_id)
page = pcgts.get_Page()