o ,hD"@sxdZddlZddlZddlZddlmmZddlm Z        dd d Z dd dZ       dddZ dS)zs Created by Jaided AI Released Date: 18/08/2022 Description: A wrapper for DBNet text detection module for EasyOCR N)DBNet皙?Fcpuc  s t|tjrt|jdkr|} n|g} tfdd| D\} } fdd| D} tt|  |} t @ | |} j | | | ||||dd\}}|rcj | | | ||||dd\}}n |}Wd||fSWd||fS1s|wY||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 csg|]}|qS) resize_image).0img canvas_sizedetectorrO/var/www/html/scripts/venv/lib/python3.10/site-packages/easyocr/detection_db.py Jsztest_net..csg|] }t|dqS))rr)np transposenormalize_image)r n_img)rrrrLsF)text_thresholdbbox_min_score bbox_min_sizemax_candidates as_polygonTN) isinstancerndarraylenshapeziptorch from_numpyarraytono_grad image2hmap hmap2bbox)imager thresholdrrrr polydevice image_arrsimagesoriginal_shapes image_tensorhmapbboxes_polysrr rtest_netsJ5      r3resnet18TcCstdtjdd|dd}|dvrtd||j|d|t|j d kr=|r..r)r r2rrrrszget_textbox..)r r*rB RuntimeErrorr?formatr3) rr'r r)r(rrrr*kwargsr1 polys_listresultrrr get_textboxs 3   r\)rrrrNFr)r4rTF)NFrrrrr) __doc__r=numpyrr torch.backends.cudnnbackendsrH DBNet.DBNetrr3rPr\rrrrs.  Z.