3
|
j|ð[ * ã @ sZ d dl mZ ddlmZ d dlZd dlZd dlZd dlZdZ ej
|
e e Z G dd dZ dS )é )Údecodeé )ÚgetSIDNz /template.pngc @ s e Zd Zd#ddZd$ddZd%d d
|
Zd d Zd d Zd&ddZdd Z dd Z
|
d'ddZ d(ddZ dd Z d d Zd d Zd!d" ZdS ))ÚPaperNú/tmpc C sd || _ || _d | _d | _|d kr(ddin|| _g | _g | _d | _|| _|d k r`| j | | j
|
d S )NÚanswer_thresholdg Ð?) ÚfilenameÚ output_pathÚinvalidÚQRDataÚsettingsÚerrorsÚwarningsÚsidÚsid_classifierÚ loadImageÚrunOcr)Úselfr r r r © r ú8/home/samo/programiranje/python/sizif-ocr/aoi_ocr/Ocr.pyÚ__init__ s
|
zPaper.__init__r c C sJ t j||| _| jd kr.| jjd d| _d S | jjdd
\| _| _d S )NzFile could not be loaded!Tr é ) Úcv2ÚimreadÚimgr Úappendr
|
ÚshapeÚ imgHeightÚimgWidth)r r Z
|
rgbchannelr r r r s
|
zPaper.loadImageú/tmp/debug_image.pngc C s t j|| j d S )N)r Úimwriter )r r r r r Ú saveImage$ s zPaper.saveImagec C s6 | j dkrd S | j | j d}| j | j d S )NTr )r
|
ÚdecodeQRandRotateÚ imgTresholdÚgenerateAnswerMatrixr! )r Z skewAngler r r r ' s
|
z Paper.runOcrc C sÞ | j dkrd S tj| jd
|
}t|}|| _t|dkrR| jjd d | _d| _ d S t|dkrx*|D ]"}|j dks||j dkrd||d< P qdW || _
|
|d j| _ |d j j }|d j j}|| jd krÚ|| jd krÚ| jd d S ) NTé r zQR code could not be found!r ÚEAN13ZQRg @é´ )r% r% )r
|
r Úblurr r Úlenr r ÚdataÚtypeÚQRDecoder ZrectÚleftÚtopr r Ú rotateAngle)r r( ÚdZddZxposZyposr r r r" 7 s*
|
|
zPaper.decodeQRandRotatec C sd t j| jd | jd f|d}t j| j|| j| jft jt jdd}|| _| jjdd
\| _| _d S )Nr g ð?éÿ )ÚflagsZ
|
borderModeZ borderValuer )r1 r1 r1 ) r ZgetRotationMatrix2Dr r Z
|
warpAffiner Z INTER_CUBICZBORDER_CONSTANTr )r ÚangleZrot_matÚresultr r r r/ O s
|
zPaper.rotateAnglec C s& t j| jddt jt jB \| _| _d S )Né r1 )r Ú thresholdr Z THRESH_BINARYZ THRESH_OTSUZthreshÚbwimg)r r r r r# d s zPaper.imgTresholdc C sà d| j }tjd| d}d}tj| jtj}x|tj|dtjd dD ]b}|d \}}}} tj |||f|| fd d tj
|
| | || }
|
|
|
rFt |
|
d krF||
|
7 }|d7 }qFW ytj || } W n d} Y nX tjd
|
| | S ) Nr1 z/tmp/debug_1.pngr g r r' iE r é
|
z/tmp/debug_2.png)r r r1 ) r7 r r ÚcvtColorr ÚCOLOR_GRAY2BGRZ HoughLinesPÚnpZpiÚlineZarctan2ÚabsZrad2deg) r ÚnegZ angle_counterr3 Úcimgr< Zx1Zy1Zx2Zy2Z
|
this_angleZskewr r r Ú getSkewAnglei s&
|
|
zPaper.getSkewAngleç333333ë?éÈ c C s t jtd}|jd d d
\}}| jd|
d d
f }t j||t j}tj||k}t j |t j
|
} g }
|
g } t |d dkrd
|
} nîtj |d } xFt |d d d
D ]0} | d | d k r¢| j| d |
|
j| d q¢W t tt |
|
| \}
|
} tj|
|
dk}tj|d}tj|
|
}
|
tj| } | | |
|
| g}xBt |d d d
D ],} t j| | | d | | d | fd d q>W t jd| || _|S )Nr r é é( Tr1 r z/tmp/debug_3.pngéÿÿÿÿrE rE rE )r r1 r1 )r r Úmarkerfilenamer r Ú matchTemplateÚTM_CCOEFF_NORMEDr; Úwherer9 r: r) ÚminÚzipr ÚsortedÚdiffÚarrayÚ rectangler ÚxMarkerLocations)r r6 ZheightÚtemplateÚwÚhÚcrop_imgÚresÚlocr? Úloc_filtered_xÚloc_filtered_yZmin_yÚptÚar r r ÚlocateUpMarkers s6
|
|
, zPaper.locateUpMarkersc C s t jtd}|jd d d
\}}| jd d
| d
f }t j||t j}tj||k}t j |t j
|
} g }
|
g } t |d dkrd
|
} nîtj |d } xFt |d d d
D ]0}|d | d kr¤| j|d |
|
j|d q¤W t tt | |
|
\} }
|
tj| dk}tj|d}tj|
|
}
|
tj| } | | |
|
| g}xBt |d d d
D ],}t j| ||d | |d | fd d q@W t jd| |d |d | j | g| _| jS )Nr r rC rD Tr1 r z/tmp/debug_4.pngrE rE rE rE )r r1 r1 )r r rF r r rG rH r; rI r9 r: r) ÚmaxrK r rL rM rN rO r r ÚyMarkerLocations)r r6 ÚwidthrQ rR rS rT rU rV r? rW rX Zmin_xZmax_xrY rZ r r r ÚlocateRightMarkers¬ s6
|
|
, zPaper.locateRightMarkersc C s¼ | j | j d}d}d}|}|| }g | _x| jd D ]z}g }xd| jd D ]V}| j|| |t||
|| |t||
f } |tj| }
|
|j |
|
| qNW | jj | q:W d S )Nr8 é é2 r r )
|
r[ r_ Ú answerMatrixr] rP r7 Úintr Z countNonZeror ) r ZroixoffZroiyoffZroiwidthZ roiheightZtotpxÚyZonelineÚxZroiZblackr r r r$ Ñ s" zPaper.generateAnswerMatrixc s j d krdS jd k r& jjdd }t jtd j td j
td j td j
f j |\}}} fdd|D fd d|D |S )
|
Nre Úsid_maskg{®Gáz¤?gR¸
ëQ¸?gffffffæ?g®Gáz®ï?c s g | ]} j j|qS r )r r )Ú.0Úe)r r r ú
|
<listcomp>õ s z*Paper.get_enhanced_sid.<locals>.<listcomp>c s g | ]} j j|qS r )r r )rg rR )r r r ri ö s )r r Úgetr r rc r r )r rf ZesÚerrÚwarnr )r r Úget_enhanced_sidè s
|
|
zPaper.get_enhanced_sidc C sØ | j d kr*| jjd d d d d d d}|S tj| j d}| jd jdkr|t|dd
t|d d t|dd
d d dS |jd }t|d t|d
|
d t|d t|d d }t |d krÐ|d |d< |S d S )Nz+Could not read QR or EAN code! Not an exam?)Úexam_idÚpage_noÚpaper_idÚ
|
faculty_idr Úutf8r r& é r r` ú,r% r )rn ro rp rq é r éûÿÿÿrE )
|
r r r Úbytesr r, r+ rc Úsplitr) )r ZretvalZqrdatar* r r r Ú get_code_dataù s.
|
|
|
|
zPaper.get_code_datac C s | j }tj| jd|d< | j|d< | j|d< t| jd | j t| j d | j
|
f|d< t| jd | j t| j d | j
|
f|d< t j | j | jd kd j |d < |d
|
d krÌ|d dkrÌ| j |d
|
< tjj| jd j| jjd d jd d d
d }tj|| j ||d< |S )Nrr Zqrr r r Z up_positionZright_positionr Z
|
ans_matrixr ro Ú.ú/z.pngÚoutput_filenamerE rE )ry rw r r r r ÚlistrP r r] r r; rN rb r Útolistrm ÚosÚpathÚjoinr r rx r r r )r r* r| r r r Úget_paper_ocr_data s
|
|
" 4zPaper.get_paper_ocr_data)NNNr )r )r )r )rA rB )rA rB )Ú__name__Ú
|
__module__Ú __qualname__r r r! r r" r/ r# r@ r[ r_ r$ rm ry r r r r r r s
|
|
|
|
|
%
|
%!r )Z pyzbar.pyzbarr Z sid_processr r Znumpyr; r Ú pkg_resourcesZ
|
markerfileÚresource_filenamer rF r r r r r Ú<module> s
|
