yhD"rdZddlZddlZddlZddlmcmZddlm Z  ddZ d dZ d dZ y) zs Created by Jaided AI Released Date: 18/08/2022 Description: A wrapper for DBNet text detection module for EasyOCR N)DBNetc t|tjrt|jdk(r|} n|g} t | D cgc]} |j | |c} \} } | D cgc]'} tj|j| d)} } tjtj| j|}tj5|j|j|}|j|| |||||d\}}|r|j|| |||||d\}}n|}ddd||fScc} wcc} w#1swYfSxYw)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 )rrF)text_thresholdbbox_min_score bbox_min_sizemax_candidates as_polygonTN) isinstancenpndarraylenshapezip resize_image transposenormalize_imagetorch from_numpyarraytono_grad image2hmap hmap2bbox)imagedetector thresholdr r r canvas_sizepolydevice image_arrsimgimagesoriginal_shapesn_img image_tensorhmapbboxes_polyss p/home/developers/rajanand/mypropertyqr-fmb-refixing-v2/venv/lib/python3.12/site-packages/easyocr/detection_db.pytest_netr.shj%$U[[)9Q)> W "Wa#bPSH$9$9#{$K#bcFOTZ [5bll833E:IF [F [##BHHV$4588@L ""rds_   $$!Xt*\#!$ !!"Grc