|
|
|
|
|
""" |
|
|
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, always=False): |
|
|
"""Resize image to given height.""" |
|
|
if (img.shape[0] > height or always): |
|
|
rat = height / img.shape[0] |
|
|
return cv2.resize(img, (int(rat * img.shape[1]), height)) |
|
|
|
|
|
return img |
|
|
|
|
|
|
|
|
def ratio(img, height=SMALL_HEIGHT): |
|
|
"""Getting scale ratio.""" |
|
|
return img.shape[0] / height |
|
|
|
|
|
|
|
|
def img_extend(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 |
