mirror of
https://github.com/qurator-spk/eynollah.git
synced 2025-10-06 14:39:55 +02:00
mbreorder: filter by .xml suffix in dir-in mode
This commit is contained in:
parent
b094a6b77f
commit
9967510327
2 changed files with 7 additions and 3 deletions
|
@ -25,6 +25,7 @@ from .utils.contour import (
|
|||
return_contours_of_image,
|
||||
return_parent_contours,
|
||||
)
|
||||
from .utils import is_xml_filename
|
||||
|
||||
DPI_THRESHOLD = 298
|
||||
KERNEL = np.ones((5, 5), np.uint8)
|
||||
|
@ -751,13 +752,13 @@ class machine_based_reading_order_on_layout:
|
|||
t0_tot = time.time()
|
||||
|
||||
if dir_in:
|
||||
self.ls_xmls = os.listdir(dir_in)
|
||||
ls_xmls = list(filter(is_xml_filename, os.listdir(dir_in)))
|
||||
elif xml_filename:
|
||||
self.ls_xmls = [xml_filename]
|
||||
ls_xmls = [xml_filename]
|
||||
else:
|
||||
raise ValueError("run requires either a single image filename or a directory")
|
||||
|
||||
for xml_filename in self.ls_xmls:
|
||||
for xml_filename in ls_xmls:
|
||||
self.logger.info(xml_filename)
|
||||
t0 = time.time()
|
||||
|
||||
|
|
|
@ -2202,3 +2202,6 @@ def is_image_filename(fname: str) -> bool:
|
|||
'.tif',
|
||||
'.tiff',
|
||||
))
|
||||
|
||||
def is_xml_filename(fname: str) -> bool:
|
||||
return fname.lower().endswith('.xml')
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue