YhD"tdZddlZddlZddlZddlmcmZddlm Z  dd Z dd Z dd Z dS)zs Created by Jaided AI Released Date: 18/08/2022 Description: A wrapper for DBNet text detection module for EasyOCR N)DBNet皙?Fcpuc nt|tjrt|jdkr|} n|g} t fd| D\} } fd| D} t jtj|  |} t j 5 | |}  | | | ||||d\}}|r! | | | ||||d\}}n|}dddn #1swxYwY||fS)a A wrapper for DBNet inference routine. Parameters ---------- image : np.ndarray or list of np.ndarray OpenCV BGR image array or list of it. detector : obj DBNet text detection object. threshold : float, optional Minimum probability for each pixel of heatmap tensor to be considered as a valid text pixel. The default is 0.2. bbox_min_score : float, optional Minimum score for each detected bounding box to be considered as a valid text bounding box. The default is 0.2. bbox_min_size : int, optional Minimum size for each detected bounding box to be considered as a valid text bounding box. The default is 3. max_candidates : int, optional Maximum number of detected bounding boxes to be considered as candidates for valid text bounding boxes. Setting to 0 implies no maximum. The default is 0. canvas_size : int, optional Target detection size. Input image will be resized such that it's shorter side is equal to the closest multiple of 32 to the provided canvas_size. If detection_size is not provided, it will be resized to the closest multiple of 32 each side. If the original size exceeds the min-/max-detection sizes (specified in DBNet_inference.yaml), it will be resized to be within the min-/max-sizes. The default is None. poly : boolean, optional If true, return the bounding boxes as find polygons, otherwise, return as rectagular. The default is False. device : str, optional Device to use. Options are "cpu" and "cuda". The default is 'cpu'. Returns ------- bboxes : list of lists List of text bounding boxes in format [left, right, top, bottom]. polys : list of lists List of polygon text bounding boxes. If argument poly is set to false, this output will also hold the value of output bboxes c<g|]}|S) resize_image).0img canvas_sizedetectors f/var/www/tools.fuzzalab.pt/emblema-extractor/venv/lib/python3.11/site-packages/easyocr/detection_db.py ztest_net..Js*#b#b#bPSH$9$9#{$K$K#b#b#bc`g|]*}tj|d+S))rr)np transposenormalize_image)r n_imgrs rrztest_net..Ls3 [ [ [5bl833E::IFF [ [ [rF)text_thresholdbbox_min_score bbox_min_sizemax_candidates as_polygonTN) isinstancerndarraylenshapeziptorch from_numpyarraytono_grad image2hmap hmap2bbox)imager thresholdrrrrpolydevice image_arrsimagesoriginal_shapes image_tensorhmapbboxes_polyss ` ` rtest_netr7sj%$$U[)9)9Q)>)> W "#b#b#b#b#bWa#b#b#bcFO [ [ [ [TZ [ [ [F#BHV$4$45588@@L ""..s>MMMrx}}##BH--55r;;MMMrr )r r6s rrzget_textbox..s( f f fRWMMuMMM f f fr)r$r.rG RuntimeErrorrDformatr7) rr+rr-r,rrrr.kwargsr5 polys_listresults r get_textboxrdsf |F HO33388%%%vhov.. 00 0 U%)2.<-:-;+6$(&,   MAzg f[e f f fF Mr)rrrrNFr)r8rTF)NFrrrrr) __doc__rBnumpyrr$torch.backends.cudnnbackendsrM DBNet.DBNetrr7rUrdr rrrjs   $$$$$$$$$!XXXXt****\#!$ !!"GGGGGGr