You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
66 lines
2.3 KiB
Python
66 lines
2.3 KiB
Python
import os
|
|
import subprocess
|
|
import sys
|
|
|
|
|
|
DOCKER_IMAGE_PREFIX = os.environ.get("DOCKER_IMAGE_PREFIX", "my_ocrd_workflow")
|
|
DOCKER_IMAGE_TAG = os.environ.get("DOCKER_IMAGE_TAG", "latest")
|
|
LOG_LEVEL = os.environ.get("LOG_LEVEL", "INFO")
|
|
|
|
|
|
sub_images = {
|
|
"ocrd": "core",
|
|
"ocrd-olena-binarize": "ocrd_olena",
|
|
"ocrd-sbb-binarize": "sbb_binarization",
|
|
"ocrd-sbb-textline-detector": "sbb_textline_detector",
|
|
"ocrd-calamari-recognize": "ocrd_calamari",
|
|
"ocrd-calamari-recognize-feat-update-calamari1": "ocrd_calamari-feat-update-calamari1",
|
|
"ocrd-tesserocr-segment-region": "ocrd_tesserocr",
|
|
"ocrd-tesserocr-segment-line": "ocrd_tesserocr",
|
|
"ocrd-tesserocr-recognize": "ocrd_tesserocr",
|
|
"ocrd-dinglehopper": "dinglehopper",
|
|
"ocrd-cis-ocropy-clip": "ocrd_cis",
|
|
"ocrd-cis-ocropy-resegment": "ocrd_cis",
|
|
"ocrd-cis-ocropy-segment": "ocrd_cis",
|
|
"ocrd-cis-ocropy-deskew": "ocrd_cis",
|
|
"ocrd-cis-ocropy-denoise": "ocrd_cis",
|
|
"ocrd-cis-ocropy-binarize": "ocrd_cis",
|
|
"ocrd-cis-ocropy-dewarp": "ocrd_cis",
|
|
"ocrd-cis-ocropy-recognize": "ocrd_cis",
|
|
"ocrd-fileformat-transform": "ocrd_fileformat",
|
|
}
|
|
|
|
|
|
def main():
|
|
argv = sys.argv.copy()
|
|
argv[0] = os.path.basename(argv[0])
|
|
|
|
sub_image = sub_images[argv[0]]
|
|
docker_image = "%s-%s:%s" % (DOCKER_IMAGE_PREFIX, sub_image, DOCKER_IMAGE_TAG)
|
|
|
|
docker_run(argv, docker_image)
|
|
|
|
|
|
def docker_run(argv, docker_image):
|
|
docker_run_options = []
|
|
docker_run_options.extend(["--rm", "-t"])
|
|
docker_run_options.extend(["--mount", "type=bind,src=%s,target=/data" % os.getcwd()])
|
|
docker_run_options.extend(["--user", "%s:%s" % (os.getuid(), os.getgid())])
|
|
docker_run_options.extend(["-e", "LOG_LEVEL=%s" % LOG_LEVEL])
|
|
|
|
# The containers currently need to run privileged to allow it to read from e.g.
|
|
# /home on SELinux secured systems such as Fedora. We might want to use udica
|
|
# instead in the future.
|
|
docker_run_options.extend(["--privileged=true"])
|
|
|
|
docker_run_options.extend([docker_image])
|
|
docker_run_options.extend(argv)
|
|
|
|
docker_run_command = ["docker", "run"] + docker_run_options
|
|
c = subprocess.run(docker_run_command)
|
|
sys.exit(c.returncode)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|