j>3ddlZddlmZddlmZddlZddlmZddlm Z ddl Z ddl mZddlmZmZgdZ d(d edeeefdedzdeeeefdededededejfdZ d(d e jdeeefdedzdeeeefdededededejfdZd eeze jzdeeefdedejfdZdedeeeeeffdZ d)de!eefde jded ed!ed"ed#ededefd$Z" d*de!eefde jded ed!ed"ed#ededefd%Z#d+d&e jde jdeeeefdzddfd'Z$dS),N)deepcopy)Any)Figure) BoundingBox Polygon4P)visualize_pagevisualize_kie_page draw_boxesrrr333333?TFgeometrypage_dimensionslabelcoloralpha linewidthfillpreserve_aspect_ratioreturnc Dt|dkstd|Drtd|\}} |\\} } \} } |rt|| x} }| | z| | z | z}} | |z| | z |z}} t j| | f||||g||Rg||R|S)aCreate a matplotlib rectangular patch for the element Args: geometry: bounding box of the element page_dimensions: dimensions of the Page in format (height, width) label: label to display when hovered color: color to draw box alpha: opacity parameter to fill the boxes, 0 = transparent linewidth: line width fill: whether the patch should be filled preserve_aspect_ratio: pass True if you passed True to the predictor Returns: a rectangular Patch rc3hK|]-}t|t pt|dkV.dS)rN) isinstancetuplelen).0elts \/var/www/html/Carbon-Document/venv/lib/python3.11/site-packages/doctr/utils/visualization.py zrect_patch..-s? a aQTZU%;%;!;!Ls3xx1} a a a a a ainvalid geometry formatrr edgecolor facecolorr)rany ValueErrormaxpatches Rectangle)rrrrrrrrheightwidthxminyminxmaxymaxwhs r rect_patchr3s2 8}}S a aX` a a aaa2333$MFE!)LT4,4,VU+++UlTD[E1!DVmdTkV3!D   t   !E/5//!E/5//    r!c(|jdkstd|\}} |dddf|rt| |n| z|dddf<|dddf|rt| |n|z|dddf<tj|||g||Rg||R|S)a Create a matplotlib polygon patch for the element Args: geometry: bounding box of the element page_dimensions: dimensions of the Page in format (height, width) label: label to display when hovered color: color to draw box alpha: opacity parameter to fill the boxes, 0 = transparent linewidth: line width fill: whether the patch should be filled preserve_aspect_ratio: pass True if you passed True to the predictor Returns: a polygon Patch rr"Nrrr#)shaper'r(r)Polygon) rrrrrrrrr+r,s r polygon_patchr9Es2 >V # #2333$MFEaaad^=R']s5&'9'9'9X]^HQQQTNaaad^=R'^s5&'9'9'9X^_HQQQTN ? !E/5//!E/5//    r!kwargsc Zt|trUt|dkrt||fi|St|dkr t t j||fi|Sn3t|t jr|jdkrt ||fi|Std)a"Create a matplotlib patch for the element Args: geometry: bounding box (straight or rotated) of the element page_dimensions: dimensions of the page in format (height, width) **kwargs: keyword arguments for the patch Returns: a matplotlib Patch rr6r5r") rrrr3r9npasarrayndarrayr7r')rrr:s rcreate_obj_patchr?ps(E""B x==A  hBB6BB B ]]a   H!5!5QQ&QQ Q Hbj ) )Bhn.F.FXAA&AAA . / //r! num_colorsc:g}tjddd|z D]~}|dz }dtjdzzdz }dtjdzzdz }|t j||||S)zGenerate num_colors color for matplotlib Args: num_colors: number of colors to generate Returns: colors: list of generated colors gv@2 gY@Z)r<arangerandomrandappendcolorsys hls_to_rgb)r@colorsihue lightness saturations r get_colorsrQsF YsE5:#5 6 6GG%i")..**R//58 29>>++b00E9  h)#y*EEFFFF Mr!rDpageimage words_onlydisplay_artefactsscale interactive add_labelsc |jdd\}} || kr || z|z |fn ||| z |zf} tj| \} } | || d|rg} |dD]}|sJt |d|dfdd d d |}| ||r| ||d D]}|sJt |d|dfd dd d |}| ||r| ||dD]}t |d|df|dd|ddddd|}| ||r| |`|r%t|ddkryt|dd |dd|dddz z zt|dd|dd |dddz z zf}n`t|dd |dddzt|dd|ddd zf}t|ddkr| j g||dRdddd|rU|dD]L}t |d|dfddd d |}| ||r| |M|r0ddl }| | d  d!d"| d#$| S)%a Visualize a full page with predicted blocks, lines and words >>> import numpy as np >>> import matplotlib.pyplot as plt >>> from doctr.utils.visualization import visualize_page >>> from doctr.models import ocr_db_crnn >>> model = ocr_db_crnn(pretrained=True) >>> input_page = (255 * np.random.rand(600, 800, 3)).astype(np.uint8) >>> out = model([[input_page]]) >>> visualize_page(out[0].pages[0].export(), input_page) >>> plt.show() Args: page: the exported Page of a Document image: np array of the page, needs to have the same shape than page['dimensions'] words_only: whether only words should be displayed display_artefacts: whether artefacts should be displayed scale: figsize of the largest windows side interactive: whether the plot should be interactive add_labels: for static plot, adds text labels on top of bounding box **kwargs: keyword arguments for the polygon patch Returns: the matplotlib figure Nrfigsizeoffblocksr dimensionsblock)rrrrrrrlinesline)rrrwordsvalue (confidence: confidence.2%))rrr)rrrrD?)sizerr artefactsartefact)rkrkrkhoveraddcd|j|jSN annotationset_textartist get_labelsels rz visualize_page..&s~G^G^_b_i_s_s_u_uGvGvr!rBpad)r7pltsubplotsimshowaxisr? add_patchrIrinttext mplcursorsCursorconnect tight_layout)rRrSrTrUrVrWrXr:r2r1rlfigaxartistsr_rectrbwordtext_locrnrs rr r ssH ;rr?DAq%&UUEAIM5 ! !A 0FDl4(((GCIIeGGENNN*')hA)A) %#j!4 #5=DIabflD LL    %t$$$'N( ( D )'$d<&8@Ficdhn T""")NN4(((W   '$&"']SS$|:LSSSS#    T"""NN4((((4 +,,11\ 21 5j9I!9LtT^O_`aObefOf9f ghh\ 21 5j9I!9LtT^O_`aObefOf9f ghh$  \ 21 5Z8H8KA8N NOO\ 21 5Z8H8KA8N NOO$ 4 +,,11% M"$"%"+ 3 B  )!+. ) )'Z(&%)    T""")NN4(((x '++33E;v;vwww Jr!c |jdd\}} || kr || z|z |fn ||| z |zf} tj| \} } | || d|rg} dt t t|d|dD}|dD]q\}}|D]i}|set|d|df|d |d d |d d ||dd|}| ||r| |jr|r0ddl }| | ddd| d| S)aVisualize a full page with predicted blocks, lines and words >>> import numpy as np >>> import matplotlib.pyplot as plt >>> from doctr.utils.visualization import visualize_page >>> from doctr.models import ocr_db_crnn >>> model = ocr_db_crnn(pretrained=True) >>> input_page = (255 * np.random.rand(600, 800, 3)).astype(np.uint8) >>> out = model([[input_page]]) >>> visualize_kie_page(out[0].pages[0].export(), input_page) >>> plt.show() Args: page: the exported Page of a Document image: np array of the page, needs to have the same shape than page['dimensions'] words_only: whether only words should be displayed display_artefacts: whether artefacts should be displayed scale: figsize of the largest windows side interactive: whether the plot should be interactive add_labels: for static plot, adds text labels on top of bounding box **kwargs: keyword arguments for the polygon patch Returns: the matplotlib figure NrrZr\ci|]\}}|| Sr)rrks r z&visualize_kie_page..Fs f f f85!a f f fr! predictionsrr^z rdrerfrgrr`rrorqcd|j|jSrsrtrys rr{z$visualize_kie_page..\r|r!rBr})r7rrrrziprQritemsr?rrIrrrr)rRrSrTrUrVrWrXr:r2r1rlrrrrLkeyrd predictionrrs rr r sH ;rr?DAq%&UUEAIM5 ! !A 0FDl4(((GCIIeGGENNN*') f fs:c$}:M6N6N+O+OQUVcQd'e'e f f fF=)//11)) U ) )J )'z*&!ggj&9ggT`Iaggg +    T""")NN4((( ) x '++33E;v;vwww Jr!boxesc |jdd\}}t|}|ddddgfxx|zcc<|ddddgfxx|zcc<|tj}|D]=}|\}} } } t j||| f| | ft|tr|ndd}>tj |tj di|dS) aDraw an array of relative straight boxes on an image Args: boxes: array of relative boxes, of shape (*, 4) image: np array, float32 or uint8 color: color to use for bounding box edges **kwargs: keyword arguments from `matplotlib.pyplot.plot` Nrrrrj)rr)r thicknessr) r7rastyper<int32tolistcv2 rectanglerrrrplot) rrSrr:r2r1_boxesboxr-r.r/r0s rr r bs ;rr?DAq e__F 111q!f9 111q!f9 ]]28 $ $F}}  !$dD$ D$<$jPU>V>V5gUU\gst   JuHvr!)Nr r rTF)TTrDTT)FTrDTTrs)%rJcopyrtypingrrmatplotlib.patchesr)matplotlib.pyplotpyplotrnumpyr<matplotlib.figurer common_typesrr__all__rrstrfloatboolr*r3r>r8r9Patchr?listrQdictr r r rr!rrs  $$$$$$$$$$$$00000000 @ @ @ (1"'...38_. :. u$ % .  .  . . . ....h(1"'((j(38_( :( u$ % (  (  ( ( ( _((((V0I% 2038_00 ] 0000234eUE.A(B#C*"yy sCx.y :yy y  y  yyy yyyy~"HH sCx.H :HH H  H  HHH HHHHVbjE#sC-