o 0j - @sddlZddlZddlmZddlZddlmZmZddl m Z m Z m Z ddl mZmZmZddlmZmZe d r>ddlZe d Gd d d eZe d d ed ejdededejf ddZe d dddZdejdejfddZdS)N)Dict)Image ImageDraw)class_requires_depsfunction_requires_depsis_dep_available) SIMFANG_FONT create_fontcreate_font_vertical) BaseCVResult JsonMixinzopencv-contrib-pythonc@sTeZdZdZdeeejffddZdeeeffddZdeeeffddZ d S) OCRResultz OCR resultreturncCsd|vr}g}g}dd|dD}dd|dD}t|D]]\}}|}tt|dd|ddd|dd |dd d}tt|dd|d dd|dd |d d d} |dksn| dkroq|||||qn|d }|d }|d d } | jdd\} } t| tj} t | }t j | | dft j dd}tdt|}t}|dr|dd}tt||D]\}\}}ztddtddtddf}t |}t|dkrDdd|D}|j||d|dt|}tdt|ddd ft|ddd f}t |ddd f|ddd f<t |ddd ftd||ddd f<ndd|D}|j||dt|tr^|d}t| | f|||j }t |t j!"d}t#||gd|d t$||}WqYqt %t | |d}t &d| d| fd}|'|dd| | f|'t || d| d| f|d}d|i}|d r|j(d!i|d j)|S)"z Converts the internal data to a PIL Image with detection and recognition results. Returns: Dict[Image.Image]: A dictionary containing two images: 'doc_preprocessor_res' and 'ocr_res_img'. text_word_regioncSg|] }|D]}|qqSr.0Zsublistitemrrg/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/inference/pipelines/ocr/result.py -s z%OCRResult._to_img..cSrrrrrrrr0s text_wordrr  rec_polys rec_textsdoc_preprocessor_resZ output_imgZdtypeZ vis_fontsrcSsg|]\}}||fqSrrrxyrrrrYs)outlinewidthfillg?NcSs g|] \}}t|t|fqSr)intr!rrrr`s )r')rrTRGBr r r model_settingsZ ocr_res_imguse_doc_preprocessorr)* enumerater)mathsqrtappendshapecv2ZcvtColorZ COLOR_BGR2RGBrZ fromarraynpZonesuint8randomseedrDrawr ziprandintarraylentolistZpolygonget_minarea_rectmaxminmean isinstancetupledraw_box_txt_finepathint32ZreshapeZ polylinesZ bitwise_andblendnewZpasteupdateZimg)selfZboxesZtxtsrridxZ word_regionZchar_box box_height box_widthimagehwZ image_rgbZimg_leftZ img_rightZ draw_leftZvis_fontboxtxtcolorZptsheightZbox_ptsimg_right_textZimg_showr-Z res_img_dictrrr_to_img"s          0".  zOCRResult._to_imgcOsDi}|d|d<|d|d<|d|d<|ddr#|djd|d<|ddkr-|d nt|d |d <|d |d <|d|d<d |vrOt|d |d <|d |d <|d |d <|d|d<t|d|d<|ddkrt|dnt|d|d<t|d|d<d|vr|d|d<|d|d<tj|g|Ri|S)axConverts the instance's attributes to a dictionary and then to a string. Args: *args: Additional positional arguments passed to the base class method. **kwargs: Additional keyword arguments passed to the base class method. Returns: Dict[str, str]: A dictionary with the instance's attributes converted to strings. input_path page_indexr-r.rres text_typeZsealdt_polystext_det_paramstextline_orientation_anglestext_rec_score_threshreturn_word_boxr rec_scoresr rec_boxestext_word_boxesr)strr5r<r_to_strrKargskwargsdatarrrrevs:               zOCRResult._to_strcOsi}|d|d<|d|d<|d|d<|ddr#|djd|d<|d|d<|d|d<|d |d <d |vr?|d |d <|d |d <|d |d <|d |d <|d|d<|d|d<|d|d<d|vrs|d|d<|d|d<tj|g|Ri|S)a[ Converts the object's data to a JSON dictionary. Args: *args: Positional arguments passed to the JsonMixin._to_json method. **kwargs: Keyword arguments passed to the JsonMixin._to_json method. Returns: Dict[str, str]: A dictionary containing the object's data in JSON format. rXrYr-r.rrZr\r]r[r^r_r`rrarrbrcr)jsonr_to_jsonrfrrrrks*                zOCRResult._to_jsonN) __name__ __module__ __qualname____doc__rrdrrWrerkrrrrrs T+rimg_sizerRrS font_pathrc Csttt|dd|dddt|dd|ddd}ttt|dd|dddt|dd|ddd}|d|krw|dkrwtd||fd}t|}|rvt|||f|}t |d||d dd n#td||fd}t|}|rt |||f|}|j ddg|d |d t ddg|dg||gd|gg} t j|t j d } t| | } t j|t jd }tj|| |tjtjdd } | S)a Draws text in a box on an image with fine control over size and orientation. Args: img_size (tuple): The size of the output image (width, height). box (np.ndarray): A 4x2 numpy array defining the corners of the box in (x, y) order. txt (str): The text to draw inside the box. font_path (str): The path to the font file to use for drawing the text. Returns: np.ndarray: An image with the text drawn in the specified box. rr rrr+r,)rrrrr)r' line_spacing)r'fontr)flagsZ borderModeZ borderValue)r)r0r1floatrrIrr9r draw_vertical_textr textr5Zfloat32r<r4ZgetPerspectiveTransformr6ZwarpPerspectiveZ INTER_NEARESTZBORDER_CONSTANT) rprRrSrqrMrNZimg_textZ draw_textruZpts1Zpts2MrVrrrrEsFDD    rErsrc CsR|\}}|D] }|j||f|||d||} | d| d} || |7}qdS)N)rur'r r)ryZgetbbox) Zdrawpositionryrur'rtr"r#charZbbox char_heightrrrrxs rxpointscCst|}ttt|ddd}d\}}}}|dd|ddkr)d}d}nd}d}|dd|ddkr>d}d}nd}d}t||||||||gtj}|S) aa Get the minimum area rectangle for the given points using OpenCV. Args: points (np.ndarray): An array of 2D points. Returns: np.ndarray: An array of 2D points representing the corners of the minimum area rectangle in a specific order (clockwise or counterclockwise starting from the top-left corner). cSs|dS)Nrr)r"rrrsz"get_minarea_rect..)key)rrrr rrr r) r4Z minAreaRectsortedlistZ boxPointsr5r<ZastyperG)r~Z bounding_boxZindex_aZindex_bZindex_cZindex_drRrrrr?s&  r?)rsr)r0r7typingrnumpyr5ZPILrrZ utils.depsrrrZ utils.fontsr r r Z common.resultr rr4rrDZndarrayrdrErxr?rrrrs8 & 7