helpers.py

# -*- coding: utf-8 -*-
"""
Helper functions for ocr project
"""
import matplotlib.pyplot as plt
import numpy as np
import cv2

SMALL_HEIGHT = 800

def implt(img, cmp=None, t=''):
    """ Show image using plt """
    plt.imshow(img, cmap=cmp)
    plt.title(t)
    plt.show()


def resize(img, height=SMALL_HEIGHT, allways=False):
    """ Resize image to given height """
    if (img.shape[0] > height or allways):
        rat = 1.0*height / img.shape[0]
        return cv2.resize(img, (int(rat*img.shape[1]), height), interpolation = cv2.INTER_CUBIC)
    return img


def ratio(img, height=SMALL_HEIGHT):
    """ Getting scale ratio """
    return img.shape[0] / height


def extendImg(img, shape):
    """ Extend 2D image (numpy array) in vertical and horizontal direction
    Shape of result image will match 'shape'
    Args:
        img: image to be extended
        shape: shape (touple) of result image
    Returns:
        Extended image
    """
    x = np.zeros(shape, np.uint8)
    x[:img.shape[0], :img.shape[1]] = img
    return x