3
|
W¬õ[Ú ã @ 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
|
d d}t|d k}t jd | t j|j t j t j
|
\}}}t |dd d} t j | d \}
|
} } } | | | |
|
|
d
|
| |
|
| d
f } | S )Nr Ú1é# r é( é é )Ú
|
iterationsé é é é2 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 r3 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 )r6 r6 r )r. r r r r r! r" r# Ú enumerater$ r ÚresizeÚformatÚstrÚpredictÚreshape)r% r& Ú
|
classifierr' r0 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
|
r4 z/tmp/sidblock2.pngr r Nr r5 r6 z/tmp/sid_no_{}.pngg ào@)r6 r6 r ) Úlenr. r r ÚshapeÚintÚranger r8 r9 r: r; r< ) r% r& r' r= r0 Zsid_lenÚ imgHeightÚimgWidthZnumWidthr> Únumr r r Úsegment_by_sid_len[ s &rH 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 è?r4 Téÿ z/tmp/sid_3rd2.pngr5 r6 z/tmp/sid_3no_{}.pngg ào@r r r )r rJ rJ r )r6 r6 r ) r r Ú MORPH_CLOSEr
|
r r ÚimreadÚ template8rB Ú matchTemplateÚTM_CCOEFF_NORMEDr ÚwhereÚcvtColorÚCOLOR_GRAY2BGRr/ ÚappendrA r# ÚdiffrC ÚarrayÚ rectangler7 r8 r9 r: r; r< )r% r& r' r= Z block_imageÚtemplater, r- ÚresÚlocÚcimgÚloc_filtered_xÚloc_filtered_yÚptÚaÚpointsr0 r> rG r r r Úsegment_by_7segmentsu sJ
|
|
, , (r` c C sr g }g }d| } | j }t| dk} tjd| tj| tjtdddd} tj| tjtdddd} ttj | d k} tjd
|
| tj| tj
|
tdddd} tj| tjtd d } ttj | d k} tj| tj
|
td d } tjd | t | |||}t |t |kst|| r,|jd t| |||}t |t |krVt| |||}|jd t||shdg}|||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
|
|
rJ éF z/tmp/enSID0.pngr r r )r r r5 z/tmp/enSID1.pngrI g à?z/tmp/enhancedSID.pngz Trying second SID algorithm.zTrying third SID algorithm.z
|
Wrong SID!)r r r r r rK r
|
r r Zthinr Z skeletonizer@ rA r3 rS r` rH )r% r= r' Zsid_warnZsid_errZimage_originalr0 r r r ÚgetSID s2
|
|
rb )r Únumpyr Zskimager r Ú pkg_resourcesZ templatefileÚresource_filenameÚ__name__rM r
|
r. r3 r@ rH r` rb r r r r Ú<module> s "8
|
