def test_crop(self):
with copy_of_directory(assets.path_to('dfki-testdata/data')) as wsdir:
ws = Workspace(self.resolver, wsdir)
pagexml_before = len(ws.mets.find_files(mimetype=MIMETYPE_PAGE))
run_processor(
OcrdAnybaseocrCropper,
resolver=self.resolver,
mets_url=str(Path(wsdir, 'mets.xml')),
input_file_grp='BIN',
output_file_grp='CROP-TEST',
parameter={},
)
ws.reload_mets()
pagexml_after = len(ws.mets.find_files(mimetype=MIMETYPE_PAGE))
self.assertEqual(pagexml_after, pagexml_before + 1)
def test_crop(self):
if not torch.cuda.is_available():
pytest.skip('CUDA is not available, cannot test dewarping')
with copy_of_directory(assets.path_to('dfki-testdata/data')) as wsdir:
ws = Workspace(self.resolver, wsdir)
pagexml_before = len(ws.mets.find_files(mimetype=MIMETYPE_PAGE))
run_processor(OcrdAnybaseocrDewarper,
resolver=self.resolver,
mets_url=str(Path(wsdir, 'mets.xml')),
input_file_grp='BIN',
output_file_grp='DEWARP-TEST',
parameter={'model_path': str(self.model_path)})
ws.reload_mets()
pagexml_after = len(ws.mets.find_files(mimetype=MIMETYPE_PAGE))
self.assertEqual(pagexml_after, pagexml_before + 1)
