adapt parser to other dataset

adapt the mung parser to other dataset (DoReMi and Musigraph)
OMR-Research · Jun 11, 2024 · 33262ba · 33262ba
1 parent 9d31de1
commit 33262ba
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 3 deletions.
diff --git a/mung/io.py b/mung/io.py
@@ -250,10 +250,20 @@ def read_nodes_from_file(filename: str) -> List[Node]:
     root = tree.getroot()
     logging.debug('XML parsed.')
     nodes = []
-    dataset = root.attrib['dataset']
-    document = root.attrib['document']
+    dataset = 'Unknown'
 
-    for i, node in enumerate(root.iter('Node')):
+    if 'dataset' in root.attrib:
+        dataset = root.attrib['dataset']
+    document = 'Unknown'
+
+    if 'document' in root.attrib:
+        document = root.attrib['document']
+    node_tag = 'Node'
+
+    if not list(root.iter(node_tag)):
+        node_tag = 'CropObject'
+
+    for i, node in enumerate(root.iter(node_tag)):
         ######################################################
         logging.debug('Parsing Node {0}'.format(i))
 

diff --git a/mung/node.py b/mung/node.py
@@ -812,6 +812,9 @@ def decode_mask_rle(mask_string: str, shape) -> Optional[numpy.ndarray]:
 
         mask_flat = numpy.zeros(shape[0]*shape[1], numpy.uint8)
         index = 0
+
+        mask_string = mask_string.strip().rstrip()
+
         for kv in mask_string.split(' '):
             k_string, v_string = kv.split(':')
             k, v = int(k_string), int(v_string)