3
|
Í÷[ ã @ st d dl Z d dlZd dlmZmZ d dlZdZeje eZ
|
dd Z dd Z dd Z d
|
d Zd d Zdd Zdd ZdS )é N)Ú
|
morphologyÚ img_as_ubytez/template-8.pngc C s t j| |ft jS )z8
|
Function greates square kernel of size x and y
|
)ÚnpZonesÚuint8)ÚxÚy© r ú@/home/samo/programiranje/python/sizif-ocr/aoi_ocr/sid_process.pyÚkernel s r
|
c C sâ |d dkrd}n|d dkr$d}nd}t j|t jtdddd}t j| t jtd d
|
dd}t|dk}t jd | t j|j t j t j
|
\}}}t |d d d} t j | d \}
|
} } } | | | |
|
|
d
|
| |
|
| d
f } | S )Nr Ú1é- r é2 é é )Ú
|
iterationsé é z/tmp/sidblock1.pngc S s
|
t j| S )N)Úcv2Z contourArea)Úctrr r r Ú<lambda>% s z#find_biggest_blob.<locals>.<lambda>)Úkeyé é éÿÿÿÿ) r Ú morphologyExÚ
|
MORPH_OPENr
|
Ú MORPH_DILATEr ÚimwriteÚ findContoursÚcopyÚ RETR_EXTERNALÚCHAIN_APPROX_SIMPLEÚsortedÚ boundingRect)ÚimageÚoriginal_imageÚsid_maskZ move_leftZimage2Zimage1Úim2ÚctrsÚhierÚ sorted_ctrsr r ÚwÚhr r r Úfind_biggest_blob s" (r- c C s0 x*t || D ] \}}|dkr ||kr dS q W dS )zÎ
|
Function compares student id number with student id mask if the recognised number is valid according to the mask
|
:param sid_no:
|
:param sid_mask:
|
:return: True if they match, else False
|
r FT)Úzip)Úsid_nor& ÚsÚesr r r Ú sid_compare, s r2 c C sò d}t | ||} tjd| tj| j tjtj\}}}t|dd d}x¤t|D ]\} }
|
tj |
|
\} } } }| |d k r|d }qR| | | |
| | |
f }t
|
|dk }tj |d}tjd
|
j | | |t |j|jd dd d }qRW |S )zæ
|
First algorithm. it segments numerals with contours. It works with numbers where individual numerals does not touch.
|
:param image:
|
:param original_image:
|
:param classifier:
|
:return: student id as a string
|
Ú z/tmp/sid_contour1.pngc S s t j| d S )Nr )r r# )r r r r r H s z%segment_by_contours.<locals>.<lambda>)r r r é é z/tmp/sid_no_{}.pngr g ào@r )r5 r5 r )r- r r r r r r! r" Ú enumerater# r ÚresizeÚformatÚstrÚpredictÚreshape)r$ r% Ú
|
classifierr& r/ r' r( r) r* Úir r r r+ r, Úroir r r Úsegment_by_contours9 s" &r? c C sÆ d}t |}t| ||} tjd| | jdd
\}}t|| }x~td|D ]p} | dd
| | | d |
f }
|
t|
|
dk }
|
tj|
|
d }
|
tjd j | |
|
|t
|
|j |
|
j dd d
|
d }qNW |S ) aR
|
Third algorithm. It trys to get biggest "blob" in the image and then it cuts it into individual numbers by force.
|
It has some problems with finding individual numbers, so some tweaking must be done!
|
|
:param image:
|
:param original_image:
|
:param sid_mask:
|
:param classifier:
|
:return: student id as a string
|
r3 z/tmp/sidblock2.pngr r Nr r4 r5 z/tmp/sid_no_{}.pngg ào@)r5 r5 r ) Úlenr- r r ÚshapeÚintÚranger r7 r8 r9 r: r; ) r$ r% r& r< r/ Zsid_lenÚ imgHeightÚimgWidthZnumWidthr= Únumr r r Úsegment_by_sid_len[ s &rG c C sH t j|t jtdddd}t|dk }t jd| t jtd}|jddd
\}}t j ||t j
|
}t j |d k} t j | t j}
|
g } g } xNt| ddd
D ]8} | d d | d d f} | j| d | j| d qW t| dkræd
|
S ttt| | \} } t j| t|d k}t j|d }t j| } t j| } | | | | g}xBt|ddd
D ],} t j|
|
| | d | | d | fdd qVW t jd |
|
d
|
}x¬tt|ddd
D ]\}} | | d | d |
| d | d |
f }t|dk }yt j|d}W n
|
d
|
S t jdj|| |t|j|j ddd d }q®W |S )aI
|
Second attempt. It dilates the image to get all 7 segments wisible as 8888888 then it does pattern matching of 8 with
|
pattern image. It works if the scaned gray level is high enough.
|
|
:param image:
|
:param original_image:
|
:param sid_mask:
|
:param classifier:
|
:return: student id number as a string
|
r é
|
)r r z/tmp/sid_3rd1.pngr Nr g è?r3 Téÿ z/tmp/sid_3rd2.pngr4 r5 z/tmp/sid_3no_{}.pngg ào@r r r )r rI rI r )r5 r5 r ) r r Ú MORPH_CLOSEr
|
r r ÚimreadÚ template8rA Ú matchTemplateÚTM_CCOEFF_NORMEDr ÚwhereÚcvtColorÚCOLOR_GRAY2BGRr. Úappendr@ r" ÚdiffrB ÚarrayÚ rectangler6 r7 r8 r9 r: r; )r$ r% r& r< Z block_imageÚtemplater+ r, ÚresÚlocÚcimgÚloc_filtered_xÚloc_filtered_yÚptÚaÚpointsr/ r= rF r r r Úsegment_by_7segmentsu sJ
|
|
, , (r_ c C s g }g }d| } | j }t| dk} tjd| tj| tjtdddd} tj| tjtdddd} tj| tjtdddd} ttj | d k} tjd
|
| tj| tj
|
tdd dd} tj| tjtd d } ttj | d k} tj| tj
|
td d } tjd| t | |||}t |t |ks.t|| rF|jd t| |||}t |t |krpt| |||}|jd t||sdg}|||fS )aÔ
|
Tries different approaches on image to get student id number. Firstly clears image of noise and then skeletonizes
|
numbers and thickens it until it gets normalized image. It sends it to the segmentation and recognition functions.
|
|
Tweak both MORPH_OPEN lines....
|
|
:param image:
|
:param classifier:
|
:param sid_mask:
|
:return: (student_id, error, warning) student id as a string, list of errors and list of warnings during the recognition
|
|
rI éF z/tmp/enSID0.pngr )r r r é r4 z/tmp/enSID1.pngr rH g à?z/tmp/enhancedSID.pngz Trying second SID algorithm.zTrying third SID algorithm.z
|
Wrong SID!)r r r r r r r
|
rJ r Zthinr Z skeletonizer? r@ r2 rR r_ rG )r$ r< r& Zsid_warnZsid_errZimage_originalr/ r r r ÚgetSID s4
|
|
rb )r Únumpyr Zskimager r Ú pkg_resourcesZ templatefileÚresource_filenameÚ__name__rL r
|
r- r2 r? rG r_ rb r r r r Ú<module> s "8
|
