3
|
í¬õ[N+ ã @ sZ d dl mZ ddlmZ d dlZd dlZd dlZd dlZdZ ej
|
e e Z G dd dZ dS )é )Údecodeé )ÚgetSIDNz/template-sq.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 sD | j dkrd S | j | j tjd| j d}| j | j d S )NTz/tmp/debug_threshold.pngr )r
|
ÚdecodeQRandRotateÚ imgTresholdr r ÚbwimgÚ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" 8 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 )r2 r2 r2 ) r ZgetRotationMatrix2Dr r Z
|
warpAffiner Z INTER_CUBICZBORDER_CONSTANTr )r ÚangleZrot_matÚresultr r r r0 P s
|
zPaper.rotateAnglec C s& t j| jddt jt jB \| _| _d S )Né r2 )r Ú thresholdr Z THRESH_BINARYZ THRESH_OTSUZthreshr$ )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 ) Nr2 z/tmp/debug_1.pngr g r r( iE r é
|
z/tmp/debug_2.png)r r r2 ) r$ r r ÚcvtColorr ÚCOLOR_GRAY2BGRZ HoughLinesPÚnpZpiÚlineZarctan2ÚabsZrad2deg) r ÚnegZ angle_counterr4 Úcimgr< Zx1Zy1Zx2Zy2Z
|
this_angleZskewr r r Ú getSkewAnglej 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 é é( Tr2 r z/tmp/debug_3.pngéÿÿÿÿrE rE rE )r r2 r2 )r r Úmarkerfilenamer r$ Ú matchTemplateÚTM_CCOEFF_NORMEDr; Úwherer9 r: r* ÚminÚzipr ÚsortedÚdiffÚarrayÚ rectangler ÚxMarkerLocations)r r7 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 jd| 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 yttt| |
|
\} }
|
W n* tjddgtjddgg| _| jS tj| dk}tj|d}tj|
|
}
|
tj| } | | |
|
| g}xBt|d d d
D ],}t j| ||d | |d | fdd q~W t jd | |d |d | j | g| _| jS )Nr r z/tmp/debug_right.pngrC r rD Tr2 z/tmp/debug_4.pngrE rE rE rE )r r2 r2 )r r rF r r$ r rG rH r; rI r9 r: r* ÚmaxrK r rL rN ÚyMarkerLocationsrM rO r )r r7 ÚwidthrQ rR rS rT rU rV r? rW rX Zmin_xZmax_xrY rZ r r r ÚlocateRightMarkers s@
|
|
, 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 r$ Ú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¸?gÍÌÌÌÌÌä?gffffffî?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
|
t|d t|d d d}t |d krÎ|d |d < |S d S )Nz+Could not read QR or EAN code! Not an exam?)Zexam_idÚpage_noZpaper_idZ
|
faculty_idr Úutf8r r' é r r` ú,r& r é r éûÿÿÿrE )
|
r r r Úbytesr r- r, rc Úsplitr* )r ZretvalZqrdatar+ r r r Ú get_code_dataÿ s0
|
|
|
|
|
|
zPaper.get_code_datac C s | j }tj| jd|d< | j|d< | j|d< t| jd | j t| jd | j f|d< t| j
|
d | j t| j
|
d | j f|d< t j | j | jd kd j |d
|
< |d d krÌ|d dkrÌ| j |d < tjj| jd j| jjdd jd d d
d }tj|| j ||d< |S )Nro Zqrr r r r Z up_positionZright_positionr Z
|
ans_matrixr rn Ú.ú/z.pngÚoutput_filenamerE rE )rv rt r r r r ÚlistrP r r r] r; rN rb r Útolistrm ÚosÚpathÚjoinr r ru r r r )r r+ ry 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" r0 r# r@ r[ r_ r% rm rv 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
|
