i5VdZddlZddlmZddlmZddlmZddl m Z Gdde Z y) zY PassportEye::Util: Geometry & math utilities Author: Konstantin Tretyakov License: MIT N)PCA)pyplot)patches) transformceZdZdZddZdZedZedZedZ ddZ d Z dd Z dd Z dd Zd ZeddZy) RotatedBoxa RotatedBox represents a rectangular box centered at (cx,cy) with dimensions width x height, rotated by angle radians counterclockwise. >>> RotatedBox.from_points([[0,0], [2,1], [0,1], [2,0]]) RotatedBox(cx=1.0, cy=0.5, width=2.0, height=1.0, angle=0.0) Nctj|tj|_||_||_||_||_y)zCreates a new RotatedBox. :param points: This parameter may be used to indicate the set of points used to create the box. dtypeN)npasarrayfloat64centerwidthheightanglepoints)selfrrrrrs ~/var/www/ITT-ServiceAPI-Services/ITT-Document-Scan/mrzscanner/mrzenv/lib/python3.12/site-packages/passporteye/util/geometry.py__init__zRotatedBox.__init__s6 jjrzz:     cdj|j|j|j|j|j S)Nz>> assert RotatedBox([2, 2], 2, 1, 0.1).rotated([1, 1], np.pi/2).approx_equal([0, 2], 2, 1, np.pi/2+0.1) r )r arraycossinr rdotTrrrrrpi)rrotation_centerrrottnew_cs rrotatedzRotatedBox.rotated6s hhu 6"&&-PU8WXY JJbjj 9suut{{Q014%T[[4::e;KPRPUPUVWPW:XYYrcL|jdz |ztjtj|jtj |jgz}|j dz |ztjtj |j tj|jgz}tj|j|jg}tj||z |z ||z|z ||z|z||z |zgS)ayConverts this box to a polygon, i.e. 4x2 array, representing the four corners starting from lower left to upper left counterclockwise. :param margin_width: The additional "margin" that will be added to the box along its width dimension (from both sides) before conversion. :param margin_height: The additional "margin" that will be added to the box along its height dimension (from both sides) before conversion. >>> RotatedBox([0, 0], 4, 2, 0).as_poly() array([[-2., -1.], [ 2., -1.], [ 2., 1.], [-2., 1.]]) >>> RotatedBox([0, 0], 4, 2, np.pi/4).as_poly() array([[-0.707..., -2.121...], [ 2.121..., 0.707...], [ 0.707..., 2.121...], [-2.121..., -0.707...]]) >>> RotatedBox([0, 0], 4, 2, np.pi/2).as_poly() array([[ 1., -2.], [ 1., 2.], [-1., 2.], [-1., -2.]]) >>> RotatedBox([0, 0], 0, 0, np.pi/2).as_poly(2, 1) array([[ 1., -2.], [ 1., 2.], [-1., 2.], [-1., -2.]]) r+) rr r,r-rr.rrrvstack)r margin_width margin_heightv_horv_vertcs ras_polyzRotatedBox.as_poly@s6A ,bhhtzz8JBFFSWS]S]L^7_.``++a--/BFF4:: setdefaultrPolygon add_patch)rmodeaxkwargspolyps rplotzRotatedBox.plot`s_3779||~ 7?1Q%=D&%('3'$" OOD +F + Qrctjdz |jz dztjz }tj|||j d|z|j d|zgd}|j ||\}}tt|j d|jdz z |z |z|z d} t|j d|jdz z|z|z|z } tt|j d|jdz z |z |z|z d} t|j d|jdz z|z|z|z } || | | | fS)aExtracts the contents of this box from a given image. For that the image is "unrotated" by the appropriate angle, and the corresponding part is extracted from it. Returns an image with dimensions height*scale x width*scale. Note that the box coordinates are interpreted as "image coordinates" (i.e. x is row and y is column), and box angle is considered to be relative to the vertical (i.e. np.pi/2 is "normal orientation") :param img: a numpy ndarray suitable for image processing via skimage. :param scale: the RotatedBox is scaled by this value before performing the extraction. This is necessary when, for example, the location of a particular feature is determined using a smaller image, yet then the corresponding area needs to be extracted from the original, larger image. The scale parameter in this case should be width_of_larger_image/width_of_smaller_image. :param margin_width: The margin that should be added to the width dimension of the box from each size. This value is given wrt actual box dimensions (i.e. not scaled). :param margin_height: The margin that should be added to the height dimension of the box from each side. :return: a numpy ndarray, corresponding to the extracted region (aligned straight). TODO: This could be made more efficient if we avoid rotating the full image and cut out the ROI from it beforehand. r+r"rT)rrresize) r r1rrrotater_compensate_rotation_shiftmaxintrr) rimgscaler9r: rotate_by img_rotatedshift_cshift_rr1r2c1c2s rextract_from_imagezRotatedBox.extract_from_imagews^(UU1Wtzz)3.ruu4 &&s)T[[QR^TYEY[_[f[fgh[ijo[oDpy}~  ::3F dkk!nt{{1}4}DeKgUVXY Z $++a.4;;q=0=@%G'Q R dkk!ntzz!|3lBEIGSTVW X $++a.4::a</,>EO P2b5"R%<((rcntj|jd|z|jd|zg}tj|}tjtj dz |j z }tj| }||z|z}|jd|jd} }tjddgd|dz g| dz |dz g| dz dgg} |j| } | dddfj} | dddfj} | dddfj} | dddfj}| | f}tj|}||z}d|jd<||j|z jjS)aThis is an auxiliary method used by extract_from_image. It is needed due to particular specifics of the skimage.transform.rotate implementation. Namely, when you use rotate(... , resize=True), the rotated image is rotated and shifted by certain amount. Thus when we need to cut out the box from the image, we need to account for this shift. We do this by repeating the computation from skimage.transform.rotate here. TODO: This makes the code uncomfortably coupled to SKImage (e.g. this logic is appropriate for skimage 0.12.1, but not for 0.11, and no one knows what happens in later versions). A solution would be to use skimage.transform.warp with custom settings, but we can think of it later. r"r) translationr+)rotationN)rrr")r r rrSimilarityTransformr1rshaper,inverseminrUparamsraveltolist)rrWrXctrtform1tform2tform3tformrowscolscornersmincminrmaxcmaxrrctform4s rrTz%RotatedBox._compensate_rotation_shiftsjj$++a.. Au0DEF..3?..a$**8LM..C4@&(YYq\399Q>> RotatedBox.from_points([[0,0]]) RotatedBox(cx=0.0, cy=0.0, width=0.0, height=0.0, angle=0.0) >>> assert RotatedBox.from_points([[0,0], [1,1], [2,2]]).approx_equal([1, 1], np.sqrt(8), 0, np.pi/4) >>> assert RotatedBox.from_points([[0,0], [1,1], [0,1], [1,0]]).approx_equal([0.5, 0.5], 1, 1, 0.0) # The angle is rather arbitrary here >>> assert RotatedBox.from_points([[0,0], [2,1], [0,1], [2,0]]).approx_equal([1, 0.5], 2, 1, 0) >>> assert RotatedBox.from_points([[0,0], [2,4], [0,4], [2,0]]).approx_equal([1, 2], 4, 2, np.pi/2) >>> assert RotatedBox.from_points([[0,0], [1,1.5], [2,0]]).approx_equal([1, 0.75], 2, 1.5, 0) >>> assert RotatedBox.from_points([[0,0], [0,1], [1,1]]).approx_equal([0.25, 0.75], np.sqrt(2), np.sqrt(2)/2, np.pi/4) r rr"g)rrrrr+)rr"rrbbmrz N z$Unknown parameter value: box_type=%s)r r rrfrrfitarctan2 components_r1r'rrhrUr/r0mean_sortedlen logical_and from_pointsr ValueError)rbox_typemrpoints_transformedllurwhh_coordnbottomtop valid_pointsrbs rrzRotatedBox.from_pointss 6F"**5 <<?a fQis3cRXY Y FJJv AMM#. c0BCbeeK uruu}  %&+QYEBEEMEBEEME[[0 VV& * VV& * "W t E 1fll1o6JbffQ]]__r"uai@177JRTUVRW`bcd`emr|BC C   /!45GG A!!B$-1RCF>>*rz)r@N)g?r)r{)__name__ __module__ __qualname____doc__rrpropertyrrr%r)r6r>rOrarT staticmethodrrrrrs L((Q Zk@.)@$;L>P>Prr) rnumpyr sklearn.decompositionr matplotlibrrErskimagerobjectrrrrrs+ %$nPnPr