| commit | author | age | ||
| 69abed | 1 | 3 |
| 5460bf | 2 | í¬õ[N+ ã @ sZ d dl mZ ddlmZ d dlZd dlZd dlZd dlZdZ ej |
| SP | 3 | e e Z G dd dZ dS )é )Údecodeé )ÚgetSIDNz/template-sq.pngc @ s e Zd Zd#ddZd$ddZd%d d |
| 69abed | 4 | Zd d Zd d Zd&ddZdd Z dd Z |
| SP | 5 | 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 |
| 6 | 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 | |
| 7 | 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 | |
| 8 | ÚshapeÚ imgHeightÚimgWidth)r r Z | |
| 9 | rgbchannelr r r r s | |
| 5460bf | 10 | 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 |
| SP | 11 | ÚdecodeQRandRotateÚ imgTresholdr r ÚbwimgÚgenerateAnswerMatrixr! )r Z skewAngler r r r ' s |
| 12 | z Paper.runOcrc C sÞ | j dkrd S tj| jd | |
| 69abed | 13 | }t|}|| _t|dkrR| jjd d | _d| _ d S t|dkrx*|D ]"}|j dks||j dkrd||d< P qdW || _ |
| 5460bf | 14 | |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 |
| SP | 15 | r Úblurr r Úlenr r ÚdataÚtypeÚQRDecoder ZrectÚleftÚtopr r Ú rotateAngle)r r) ÚdZddZxposZyposr r r r" 8 s* |
| 69abed | 16 | |
| SP | 17 | zPaper.decodeQRandRotatec C sd t j| jd | jd f|d}t j| j|| j| jft jt jdd}|| _| jjdd \| _| _d S )Nr g ð?éÿ )ÚflagsZ |
| 5460bf | 18 | borderModeZ borderValuer )r2 r2 r2 ) r ZgetRotationMatrix2Dr r Z |
| SP | 19 | warpAffiner Z INTER_CUBICZBORDER_CONSTANTr )r ÚangleZrot_matÚresultr r r r0 P s |
| 20 | 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 | |
| 69abed | 21 | | | || } |
| SP | 22 | | |
| 23 | rFt | | |
| 24 | d krF|| | |
| 25 | 7 }|d7 }qFW ytj || } W n d} Y nX tjd | |
| 5460bf | 26 | | | S ) Nr2 z/tmp/debug_1.pngr g r r( iE r é |
| SP | 27 | z/tmp/debug_2.png)r r r2 ) r$ r r ÚcvtColorr ÚCOLOR_GRAY2BGRZ HoughLinesPÚnpZpiÚlineZarctan2ÚabsZrad2deg) r ÚnegZ angle_counterr4 Úcimgr< Zx1Zy1Zx2Zy2Z |
| 28 | this_angleZskewr r r Ú getSkewAnglej s& | |
| 69abed | 29 | |
| SP | 30 | 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 |
| 31 | } g } | |
| 32 | g } t |d dkrd | |
| 33 | } nîtj |d } xFt |d d d D ]0} | d | d k r¢| j| d | | |
| 34 | j| d q¢W t tt | | |
| 35 | | \} | |
| 36 | } tj| | |
| 37 | dk}tj|d}tj| | |
| 38 | } | |
| 39 | tj| } | | | | |
| 5460bf | 40 | | 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 |
| 69abed | 41 | |
| 5460bf | 42 | , zPaper.locateUpMarkersc C sÜ t jtd}|jd d d |
| SP | 43 | \}}| jd d | d f }t jd| t j||t j}tj ||k}t j |
| 44 | |t j } g } | |
| 45 | g } t |d dkrd } n tj |d } xFt|d d d D ]0}|d | d kr²| j|d | | |
| 46 | j|d q²W yttt| | | |
| 69abed | 47 | \} } |
| 5460bf | 48 | W n* tjddgtjddgg| _| jS tj| dk}tj|d}tj| |
| 69abed | 49 | } |
| 5460bf | 50 | tj| } | | | |
| SP | 51 | | 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@ |
| 69abed | 52 | |
| SP | 53 | , zPaper.locateRightMarkersc C s¼ | j | j d}d}d}|}|| }g | _x| jd D ]z}g }xd| jd D ]V}| j|| |t|| || |t|| f } |tj| } |
| 54 | |j | | |
| 55 | | qNW | jj | q:W d S )Nr8 é é2 r r ) | |
| 5460bf | 56 | 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 ) |
| SP | 57 | Nre Úsid_maskg{®Gáz¤?gR¸ ëQ¸?gÍÌÌÌÌÌä?gffffffî?c s g | ]} j j|qS r )r r )Ú.0Úe)r r r ú |
| 58 | <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 | |
| 69abed | 59 | |
| 5460bf | 60 | 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 |
| SP | 61 | 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 |
| 62 | faculty_idr Úutf8r r' é r r` ú,r& r é r éûÿÿÿrE ) | |
| 63 | r r r Úbytesr r- r, rc Úsplitr* )r ZretvalZqrdatar+ r r r Ú get_code_dataÿ s0 | |
| 69abed | 64 | |
| SP | 65 | |
| 5460bf | 66 | |
| SP | 67 | |
| 68 | ||
| 69 | zPaper.get_code_datac C s | j }tj| jd|d< | j|d< | j|d< t| jd | j t| jd | j f|d< t| j | |
| 70 | d | j t| j | |
| 71 | d | j f|d< t j | j | jd kd j |d | |
| 72 | < |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 | |
| 73 | 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 | |
| 69abed | 74 | |
| SP | 75 | " 4zPaper.get_paper_ocr_data)NNNr )r )r )r )rA rB )rA rB )Ú__name__Ú |
| 5460bf | 76 | __module__Ú __qualname__r r r! r r" r0 r# r@ r[ r_ r% rm rv r r r r r r s |
| 69abed | 77 | |
| SP | 78 | |
| 5460bf | 79 | |
| SP | 80 | |
| 69abed | 81 | % |
| 5460bf | 82 | *"r )Z pyzbar.pyzbarr Z sid_processr r Znumpyr; r| Ú pkg_resourcesZ |
| SP | 83 | markerfileÚresource_filenamer rF r r r r r Ú<module> s |
