qurator-spk/sbb_textline_detection
1
0
Fork 0
mirror of https://github.com/qurator-spk/sbb_textline_detection.git synced 2025-10-31 01:14:17 +01:00
Code Issues Releases Wiki Activity

processor: self.resolve_resource model

Browse source
This commit is contained in:
Konstantin Baierer 2021-01-28 12:49:47 +01:00
parent 451b71f906
commit 5c7fd26883
2 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
qurator/sbb_textline_detector/ocrd_cli.py
View file

@ -46,6 +46,7 @@ class OcrdSbbTextlineDetectorRecognize(Processor):
log = getLogger('processor.OcrdSbbTextlineDetectorRecognize') log = getLogger('processor.OcrdSbbTextlineDetectorRecognize')
assert_file_grp_cardinality(self.input_file_grp, 1) assert_file_grp_cardinality(self.input_file_grp, 1)
assert_file_grp_cardinality(self.output_file_grp, 1) assert_file_grp_cardinality(self.output_file_grp, 1)
model = self.resolve_resource(self.parameter['model'])
for (n, input_file) in enumerate(self.input_files): for (n, input_file) in enumerate(self.input_files):
page_id = input_file.pageId or input_file.ID page_id = input_file.pageId or input_file.ID
@ -73,7 +74,6 @@ class OcrdSbbTextlineDetectorRecognize(Processor):
page_image.save(image_file) page_image.save(image_file)
# Segment the image # Segment the image
model = self.parameter['model']
x = textline_detector(image_file, tmp_dirname, file_id, model) x = textline_detector(image_file, tmp_dirname, file_id, model)
x.run() x.run()

2
requirements.txt
View file

@ -9,5 +9,5 @@ scikit-learn
numpy == 1.18.* # XXX for tensorflow-gpu 1.15 numpy == 1.18.* # XXX for tensorflow-gpu 1.15
tensorflow-gpu == 1.15.* tensorflow-gpu == 1.15.*
scipy scipy
ocrd >= 2.18.0 ocrd >= 2.22.3
shapely >= 1.7.1 shapely >= 1.7.1