4.py

#!/usr/bin/env python3

# local/process_data.py data/local

import argparse
import os
import xml.dom.minidom as minidom

import matplotlib.pyplot as plt
import matplotlib.patches as patches
from PIL import Image
import numpy as np

parser = argparse.ArgumentParser(description="""Creates text, utt2spk
                                                and images.scp files.""")
parser.add_argument('database_path', type=str,
                    help='Path to the downloaded (and extracted) IAM data')
args = parser.parse_args()

xml_file_loc = os.path.join(args.database_path,
                              'madcat_sample')

im = Image.open('/Users/ashisharora/madcat_sample/CNS_CMN_20070126.0303_002_LDC0523.tif')

# Create figure and axes
fig,ax = plt.subplots(1)
ax.imshow(im)

print("Processing data...")

xml_path = os.path.join(xml_file_loc , 'CNS_CMN_20070126.0303_002_LDC0523.gedi' + '.xml')
doc = minidom.parse(xml_path)

DL_ZONE = doc.getElementsByTagName('DL_ZONE')
previous_id="-1"
start="true"
region_list = list()
for node in DL_ZONE:
    id=node.getAttribute('lineID')
    if id=="":
        continue
    if previous_id != id and start=="false":
        print(previous_id, id)
        break
    col=node.getAttribute('col')
    row=node.getAttribute('row')
    width=node.getAttribute('width')
    height=node.getAttribute('height')

    col_down = int(col)+int(width)
    row_right = int(row)+int(height)
    box = (int(col), int(row), col_down, row_right)
    region = im.crop(box) # cropping a box

    rect1 = patches.Rectangle((col, row), width, height, linewidth=1, edgecolor='w', facecolor='none')
    ax.add_patch(rect1)
    region_list.append(region)
    previous_id = id
    start="false"

result_width=0
result_height=-1
for x in region_list:
    (width, height) = x.size
    result_width += width
    result_height = max(result_height, height)

result = Image.new('RGB', (result_width, result_height))

box_width=0
for x in region_list:
    result.paste(im=x, box=(box_width,0))
    (width, height) = x.size
    box_width = box_width + width

result.show()
plt.show()