Ocropus nlbin threshold extension

This commit is contained in:
Inga Kirschnick 2022-11-10 12:14:03 +01:00
parent 1faa98b4e8
commit f4f0628b60
3 changed files with 11 additions and 1 deletions

View File

@ -3,6 +3,7 @@ from flask_wtf import FlaskForm
from flask_wtf.file import FileField, FileRequired from flask_wtf.file import FileField, FileRequired
from wtforms import ( from wtforms import (
BooleanField, BooleanField,
IntegerRangeField,
MultipleFileField, MultipleFileField,
SelectField, SelectField,
StringField, StringField,
@ -49,6 +50,9 @@ class CreateTesseractOCRPipelineJobForm(CreateJobBaseForm):
binarization = BooleanField('Binarization') binarization = BooleanField('Binarization')
pdf = FileField('File', validators=[FileRequired()]) pdf = FileField('File', validators=[FileRequired()])
model = SelectField('Model', validators=[InputRequired()]) model = SelectField('Model', validators=[InputRequired()])
ocropus_nlbin_threshold = IntegerRangeField(
render_kw={'tooltips':'false', 'min': 0, 'max': 1, 'step': 0.1, 'start': [0.5]}
)
def validate_binarization(self, field): def validate_binarization(self, field):
service_info = SERVICES['tesseract-ocr-pipeline']['versions'][self.version.data] service_info = SERVICES['tesseract-ocr-pipeline']['versions'][self.version.data]

View File

@ -78,7 +78,8 @@ def tesseract_ocr_pipeline():
service=service_name, service=service_name,
service_args={ service_args={
'binarization': form.binarization.data, 'binarization': form.binarization.data,
'model': hashids.decode(form.model.data) 'model': hashids.decode(form.model.data),
'ocropus_nlbin_threshold': form.ocropus_nlbin_threshold.data
}, },
service_version=form.version.data, service_version=form.version.data,
user=current_user user=current_user

View File

@ -83,6 +83,11 @@
</label> </label>
</div> </div>
</div> </div>
<div class="col s12"><p>&nbsp;</p></div>
<div class="col s9">
<p>Intensity</p>
<p class="range-field">{{ form.ocropus_nlbin_threshold() }}</p>
</div>
<div class="col s12"><p>&nbsp;</p></div> <div class="col s12"><p>&nbsp;</p></div>
<div class="col s12 divider"></div> <div class="col s12 divider"></div>
<div class="col s12"><p>&nbsp;</p></div> <div class="col s12"><p>&nbsp;</p></div>