sporada/ocr
1
0
Fork 0
mirror of https://gitlab.ub.uni-bielefeld.de/sfb1288inf/ocr.git synced 2025-01-30 21:09:00 +00:00
Code Issues Projects Releases Wiki Activity

Sort files in output.

Browse Source
This commit is contained in:
Patrick Jentsch 2019-04-15 09:47:30 +02:00
parent eb6327aed3
commit d84db585fa
1 changed files with 27 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
ocr
View File

@ -84,7 +84,14 @@ class OCRWorkflow(WorkflowRunner):
create_output_directories_job_number = 0 create_output_directories_job_number = 0
for job in self.jobs: for job in self.jobs:
create_output_directories_job_number += 1 create_output_directories_job_number += 1
cmd = 'mkdir -p "%s"' % (os.path.join(job["output_dir"], "tmp")) cmd = 'mkdir -p "%s"' % (
os.path.join(job["output_dir"], "tmp", "binarized"),
os.path.join(job["output_dir"], "tmp", "hocr"),
os.path.join(job["output_dir"], "tmp", "normalized"),
os.path.join(job["output_dir"], "tmp", "pdf"),
os.path.join(job["output_dir"], "tmp", "tiff"),
os.path.join(job["output_dir"], "tmp", "txt")
)
create_output_directories_jobs.append(self.addTask(label="create_output_directories_job_-_%i" % (create_output_directories_job_number), command=cmd)) create_output_directories_jobs.append(self.addTask(label="create_output_directories_job_-_%i" % (create_output_directories_job_number), command=cmd))
### ###
@ -226,7 +233,25 @@ class OCRWorkflow(WorkflowRunner):
### ###
cleanup_jobs = [] cleanup_jobs = []
cleanup_job_counter = 0 cleanup_job_counter = 0
if not self.keepIntermediates: if self.keepIntermediates:
for job in self.jobs:
cleanup_job_counter += 1
cmd = 'mv "%s"/*.bin.png "%s" && mv "%s"/*.hocr "%s" && mv "%s"/*.nrm.png "%s" && mv "%s"/*.pdf "%s" && mv "%s"/*.tif "%s" && mv "%s"/*.txt "%s"' % (
os.path.join(job["output_dir"], "tmp"),
os.path.join(job["output_dir"], "tmp", "binarized"),
os.path.join(job["output_dir"], "tmp"),
os.path.join(job["output_dir"], "tmp", "hocr"),
os.path.join(job["output_dir"], "tmp"),
os.path.join(job["output_dir"], "tmp", "normalized"),
os.path.join(job["output_dir"], "tmp"),
os.path.join(job["output_dir"], "tmp", "pdf"),
os.path.join(job["output_dir"], "tmp"),
os.path.join(job["output_dir"], "tmp", "tiff"),
os.path.join(job["output_dir"], "tmp"),
os.path.join(job["output_dir"], "tmp", "txt")
)
cleanup_jobs.append(self.addTask(label="cleanup_job_-_%i" % (cleanup_job_counter), command=cmd, dependencies=hocr_to_tei_jobs + pdf_merge_jobs + txt_merge_jobs))
else:
for job in self.jobs: for job in self.jobs:
cleanup_job_counter += 1 cleanup_job_counter += 1
cmd = 'rm -r "%s"' % ( cmd = 'rm -r "%s"' % (