Path: blob/master/Model-1/ocr/normalization.pyc
427 views
�
jF�Zc @ s� d Z d d l Z d d l Z d d l Z d d l Td � Z d e d d � Z e d d � Z d
� Z
d � Z d d d
� � YZ d � Z
e e d e d � Z d d � Z e e d � Z d S( s{
Include functions for normalizing images of words and letters
Main functions: imageNorm, letterNorm, imageStandardization
i����Ni ( t *c C s7 | t j | � t t j | � d t j | j � � S( sB Image standardization same as tf.image.per_image_standardization g �?( t npt meant maxt stdt matht sqrtt size( t image( ( s ocr/normalization.pyt imageStandardization
s i2 i c C s t j | | d t j � \ } } d } d } | j d } | j d } xJ t | j d � D]5 }
t j | |
d d � f � d k r[ |
} Pq[ q[ WxT t t | j d � � D]9 }
t j | |
d d � f � d k r� |
d } Pq� q� WxJ t | j d � D]5 }
t j | d d � |
f � d k r� |
} Pq� q� WxT t t | j d � � D]9 }
t j | d d � |
f � d k rR|
d } PqRqRW| d k r�t | | | � | | � f | t � } n | | | � | | � f } | rt j
| d d | | t j d d d d g �S| S( s: Crop and add border to word image of letter segmentation i� i i Nt value( t cv2t thresholdt
THRESH_TOZEROt shapet rangeR t
count_nonzerot reversedt resizet Truet copyMakeBordert BORDER_CONSTANT( t imgt heightR t bordert
borderSizet rett x0t y0t x1t y1t i( ( s ocr/normalization.pyt
cropAddBorder s: !
% %
% %
+ c C s t j | d d d d �} t j | d t j d d � } | d k rd } t j d � | D� � } | d k r� t j g | D]< } | d d d
k s� | d d d k r~ | d d ^ q~ � } n | d k r| d
k s� | d k rt | | | � } qn t | | d | | � S(
s+ Detect the angle for tiltByAngle function i2 i� t apertureSizei i i� i i c s s= | ]3 } | d d d k s1 | d d d k r d Vq d S( i i gffffff�?g������@N( ( t .0t l( ( s ocr/normalization.pys <genexpr>A s gffffff�?g������@N(
R t Cannyt
HoughLinesR t pit Nonet sumR t tiltByAngleR ( R R R R t edgest linest meanAnglet numLinesR# ( ( s ocr/normalization.pyt wordTilt9 s U$c C s t j | � | } t | d � } t j d d g d | g | | g | d g g � } | d k r� t j d d g | | g | | | g | d g g � } n8 t j | d g d | g | | g | | d g g � } t j | | � } t j | | t | t | � � | f � S( s Tilt the image by given angle i ( R t tant lent float32R t getPerspectiveTransformt warpPerspectivet intt abs( R t angleR t distt widtht sPointst tPointst M( ( s ocr/normalization.pyR) K s 3
c C se t j | t j d d � } t j | t j d d � } t j | | � } d | | d k <t j | � S( s The Sobel Operatori i i� ( R t Sobelt CV_16SR t hypott uint8( t channelt sobelXt sobelYt sobel( ( s ocr/normalization.pyt sobelDetectb s
t HysterThreshc B s, e Z d � Z d � Z d � Z d � Z RS( c C s� d | } | t j | � t j | � t j | � d } t j | j � d d d g � \ } } t j | � d | _ t j | � d | _ d | j | _ | | _ t j
| j d | j �| _
d S( Ni� i i iA i- t dtype( R t minR t histogramt ravelt argmaxt hight lowt diffR t zerosR RF t im( t selfR t histt bins( ( s ocr/normalization.pyt __init__m s
1' c C s | j � t j | j � S( N( t hysterR R? RO ( RP ( ( s ocr/normalization.pyt getImagey s
c C s | j j \ } } x� t | d | d � D]� } x� t | d | d � D]� } | | k oe d k n rH | | k o� d k n rH | j | | f d k rH | j | j | | f k o� | j k n rH | j | | f | j | j | | f <| j | | � qH qH Wq* Wd S( Ni i i ( R R R RO RK RL RM t
hyster_rec( RP t rt ct ht wt rit ci( ( s ocr/normalization.pyRV } s /'c C s� | j j \ } } x� t | � D]r } xi t | � D][ } | j | | f | j k r2 d | j | | f <d | j | | f <| j | | � q2 q2 Wq Wd S( Ni� ( R R R RK RO RV ( RP RW RX R[ R\ ( ( s ocr/normalization.pyRT � s ( t __name__t
__module__RS RU RV RT ( ( ( s ocr/normalization.pyRE l s c C s= t j | t j � } t j | d d d � } t | � j � S( s3 Word normalization using hystheresis thresholding i i i ( R t cvtColort COLOR_RGB2GRAYt bilateralFilterRE RU ( R t grayR ( ( s ocr/normalization.pyt
hystImageNorm� s c C s� t | | t � } | r' t | � } nZ t j | d d d � } t j | t j � } t | � } t j | d d t j � \ }
} | r� t
| | | | � S| S( s?
Preprocess image
=> resize, get edges, tilt world
i i i2 i� ( R R Rc R Ra R_ R` RD R R
R. ( R R R t tiltR t hystNormt thR Rb R* R ( ( s ocr/normalization.pyt imageNorm� s !i8 c C s� | j d | j d k ra t | � | j d } t j | t | | j d � | f d t j �St | � | j d } t j | | t | | j d � f � S| S( s/ Resize bigger side of the image to given size i i t
interpolation( R t floatR R R4 t INTER_CUBIC( R R t rat( ( s ocr/normalization.pyt resizeLetter� s 0'c C so | rM | j d d k rM | j d d k rM t | d d d d d t �} n | } | j d d k r� | j d d k r� t | � } n t j d
t j � } d d g } | j d | j d k r� t | j d | j d d � d g } n( d t | j d | j d d � g } | | | d | d | j d � | d | d | j d � f <| rk| | j f S| S( s Preprocess an image - crop i i R R iP R i@ i i ( i@ i@ ( R R t FalseRl R RN R? R4 ( R t is_thresht dimt resizedt resultt offset( ( s ocr/normalization.pyt
letterNorm� s ,!& +(B
( ( t __doc__t numpyR R R t helpersR R R R. R) RD RE Rc Rm Rg Rl Rs ( ( ( s ocr/normalization.pyt <module> s
'
&