Path: blob/master/Model-1/ocr/helpers.pyc
427 views
�
�>�Zc @ sj d Z d d l j Z d d l Z d d l Z d Z d d d � Z e e
d � Z e d � Z d � Z
d S( s"
Helper functions for ocr project
i����Ni t c C s. t j | d | �t j | � t j � d S( s Show image using plt t cmapN( t pltt imshowt titlet show( t imgt cmpt t( ( s ocr/helpers.pyt implt s
c C sb | j d | k s | r^ d | | j d } t j | t | | j d � | f d t j �S| S( s Resize image to given height i g �?i t
interpolation( t shapet cv2t resizet intt INTER_CUBIC( R t heightt allwayst rat( ( s ocr/helpers.pyR
s 0c C s | j d | S( s Getting scale ratio i ( R ( R R ( ( s ocr/helpers.pyt ratio s c C sC t j | t j � } | | d | j d � d | j d � f <| S( s� 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
Ni i ( t npt zerost uint8R ( R R t x( ( s ocr/helpers.pyt extendImg s *( t __doc__t matplotlib.pyplott pyplotR t numpyR R t SMALL_HEIGHTt NoneR t FalseR
R R ( ( ( s ocr/helpers.pyt <module> s