| commit | author | age | ||
| 69abed | 1 | 3 |
| SP | 2 | j|ð[ * ã @ sZ d dl mZ ddlmZ d dlZd dlZd dlZd dlZdZ ej |
| 3 | e e Z G dd dZ dS )é )Údecodeé )ÚgetSIDNz /template.pngc @ s e Zd Zd#ddZd$ddZd%d d | |
| 4 | Zd d Zd d Zd&ddZdd Z dd Z | |
| 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 | |
| 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 s6 | j dkrd S | j | j d}| j | j d S )NTr )r | |
| 11 | ÚdecodeQRandRotateÚ imgTresholdÚgenerateAnswerMatrixr! )r Z skewAngler r r r ' s | |
| 12 | z Paper.runOcrc C sÞ | j dkrd S tj| jd | |
| 13 | }t|}|| _t|dkrR| jjd d | _d| _ d S t|dkrx*|D ]"}|j dks||j dkrd||d< P qdW || _ | |
| 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 | |
| 15 | r Úblurr r Úlenr r ÚdataÚtypeÚQRDecoder ZrectÚleftÚtopr r Ú rotateAngle)r r( ÚdZddZxposZyposr r r r" 7 s* | |
| 16 | ||
| 17 | zPaper.decodeQRandRotatec C sd t j| jd | jd f|d}t j| j|| j| jft jt jdd}|| _| jjdd \| _| _d S )Nr g ð?éÿ )ÚflagsZ | |
| 18 | borderModeZ borderValuer )r1 r1 r1 ) r ZgetRotationMatrix2Dr r Z | |
| 19 | warpAffiner Z INTER_CUBICZBORDER_CONSTANTr )r ÚangleZrot_matÚresultr r r r/ O s | |
| 20 | 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 | |
| 21 | | | || } | |
| 22 | | | |
| 23 | rFt | | |
| 24 | d krF|| | |
| 25 | 7 }|d7 }qFW ytj || } W n d} Y nX tjd | |
| 26 | | | S ) Nr1 z/tmp/debug_1.pngr g r r' iE r é | |
| 27 | z/tmp/debug_2.png)r r r1 ) r7 r r ÚcvtColorr ÚCOLOR_GRAY2BGRZ HoughLinesPÚnpZpiÚlineZarctan2ÚabsZrad2deg) r ÚnegZ angle_counterr3 Úcimgr< Zx1Zy1Zx2Zy2Z | |
| 28 | this_angleZskewr r r Ú getSkewAnglei s& | |
| 29 | ||
| 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| } | | | | |
| 40 | | 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 | |
| 41 | ||
| 42 | , zPaper.locateUpMarkersc C s t jtd}|jd d d \}}| jd d | d f }t j||t j}tj||k}t j |t j | |
| 43 | } g } | |
| 44 | g } t |d dkrd | |
| 45 | } nîtj |d } xFt |d d d D ]0}|d | d kr¤| j|d | | |
| 46 | j|d q¤W t tt | | | |
| 47 | \} } | |
| 48 | tj| dk}tj|d}tj| | |
| 49 | } | |
| 50 | tj| } | | | | |
| 51 | | 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 | |
| 52 | ||
| 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 ) | |
| 56 | 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 ) | |
| 57 | 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 ú | |
| 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 | |
| 59 | ||
| 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 | |
| 61 | 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Ú | |
| 62 | faculty_idr Úutf8r r& é r r` ú,r% r )rn ro rp rq é r éûÿÿÿrE ) | |
| 63 | r r r Úbytesr r, r+ rc Úsplitr) )r ZretvalZqrdatar* r r r Ú get_code_dataù s. | |
| 64 | ||
| 65 | ||
| 66 | ||
| 67 | zPaper.get_code_datac C s | j }tj| jd|d< | j|d< | j|d< t| jd | j t| j d | j | |
| 68 | f|d< t| jd | j t| j d | j | |
| 69 | f|d< t j | j | jd kd j |d < |d | |
| 70 | d krÌ|d dkrÌ| j |d | |
| 71 | < tjj| jd j| jjd d jd d d d }tj|| j ||d< |S )Nrr Zqrr r r Z up_positionZright_positionr Z | |
| 72 | 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 | |
| 73 | ||
| 74 | " 4zPaper.get_paper_ocr_data)NNNr )r )r )r )rA rB )rA rB )Ú__name__Ú | |
| 75 | __module__Ú __qualname__r r r! r r" r/ r# r@ r[ r_ r$ rm ry r r r r r r s | |
| 76 | ||
| 77 | ||
| 78 | ||
| 79 | ||
| 80 | % | |
| 81 | %!r )Z pyzbar.pyzbarr Z sid_processr r Znumpyr; r Ú pkg_resourcesZ | |
| 82 | markerfileÚresource_filenamer rF r r r r r Ú<module> s | |
