Fix line_width attribute in XMLConverter class

stefanklut · Jan 12, 2024 · ddecaf9 · ddecaf9
1 parent 65b05c6
commit ddecaf9
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/page_xml/xml_converter.py b/page_xml/xml_converter.py
@@ -471,7 +471,7 @@ def to_sem_seg(
             sem_seg = self.build_baseline_sem_seg(
                 gt_data,
                 image_shape,
-                line_width=xml_regions.line_width,
+                line_width=self.xml_regions.line_width,
             )
             return sem_seg
         elif self.xml_regions.mode == "top_bottom":
@@ -485,28 +485,28 @@ def to_sem_seg(
             sem_seg = self.build_start_sem_seg(
                 gt_data,
                 image_shape,
-                line_width=xml_regions.line_width,
+                line_width=self.xml_regions.line_width,
             )
             return sem_seg
         elif self.xml_regions.mode == "end":
             sem_seg = self.build_end_sem_seg(
                 gt_data,
                 image_shape,
-                line_width=xml_regions.line_width,
+                line_width=self.xml_regions.line_width,
             )
             return sem_seg
         elif self.xml_regions.mode == "separator":
             sem_seg = self.build_separator_sem_seg(
                 gt_data,
                 image_shape,
-                line_width=xml_regions.line_width,
+                line_width=self.xml_regions.line_width,
             )
             return sem_seg
         elif self.xml_regions.mode == "baseline_separator":
             sem_seg = self.build_baseline_separator_sem_seg(
                 gt_data,
                 image_shape,
-                line_width=xml_regions.line_width,
+                line_width=self.xml_regions.line_width,
             )
             return sem_seg
         elif self.xml_regions.mode == "text_line":

diff --git a/page_xml/xml_regions.py b/page_xml/xml_regions.py
@@ -46,6 +46,7 @@ def __init__(
         else:
             assert line_width is not None
 
+            self._regions = self._build_regions()
             self.line_width = line_width
 
     # REVIEW is this the best place for this