File size: 1,009 Bytes
0c84ee8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 |
# -*- 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, 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 |