U 9%e`*@sdZddlmZddlmZmZddlZddlm Z ddl m Z ddl m Z mZmZerbddlZerpddlZGd d d e ZdS) z Processor class for SAM. )deepcopy)OptionalUnionN)ProcessorMixin) BatchEncoding) TensorTypeis_tf_availableis_torch_availablecseZdZdZdgZdZfddZdeee e fe ddd Z dd d Z d dZdeejejdddZdddZeddZddZZS) SamProcessora Constructs a SAM processor which wraps a SAM image processor and an 2D points & Bounding boxes processor into a single processor. [`SamProcessor`] offers all the functionalities of [`SamImageProcessor`]. See the docstring of [`~SamImageProcessor.__call__`] for more information. Args: image_processor (`SamImageProcessor`): An instance of [`SamImageProcessor`]. The image processor is a required input. image_processorZSamImageProcessorcs,t||j|_d|_|jjd|_dS)Ni longest_edge)super__init__r Zcurrent_processorpoint_pad_valuesize target_size)selfr  __class__e/var/www/html/Darija-Ai-API/env/lib/python3.8/site-packages/transformers/models/sam/processing_sam.pyr2s zSamProcessor.__init__N)return_tensorsreturnc Ks`|j|fd|i|}|d}t|dr0|}|j|||d\}}}|j||||||d}|S)z This method uses [`SamImageProcessor.__call__`] method to prepare image(s) for the model. It also prepares 2D points and bounding boxes for the model if they are provided. roriginal_sizesnumpy) input_points input_labels input_boxes)rrrr)r hasattrr_check_and_preprocess_points_normalize_and_convert) rZimagesrrrrkwargsencoding_image_processorrrrr__call__8s0    zSamProcessor.__call__ptcsddk r~ttkr.fddDnfddtDtfddDst|dk rt|\}t|dk rt|}|dk rtt|krfdd|D}nfddt|D}t|}|dk r`|dkrt|}t|jd kr| d n|}n4|d krRt |}t|jd krNt |d n|}| d |idk r|dkrttjd kr d nn4|d krt tjd krt d n| di|dk r`|dkrt|}t|jd kr| d n|}n4|d krRt |}t|jd krNt |d n|}| d|i|S)Ncs g|]}j|dqSr_normalize_coordinatesr.0pointrrrr msz7SamProcessor._normalize_and_convert..cs g|]\}}j||qSrr')r*r+ original_sizerrrr-qsc3s|]}|jdjkVqdS)rNshaper))rrr vsz6SamProcessor._normalize_and_convert..cs$g|]}jj|dddqS)rTis_bounding_boxr'r*boxr,rrr-scs$g|]\}}jj||ddqS)Tr3r')r*r6r.r/rrr-sr%rtfrrr)lenzipall_pad_points_and_labelsnparraytorchZ from_numpyr1Z unsqueezer8Zconvert_to_tensorZ expand_dimsupdate)rr#rrrrrr)rrrrr!bs`                             z#SamProcessor._normalize_and_convertcCstdd|D}g}t|D]d\}}|jd|krxtj|t||jddf|jgdd}t|||jg||<||q|}||fS)zh The method pads the 2D points and labels to the maximum number of points in the batch. cSsg|]}|jdqSr&r0r)rrrr-sz7SamProcessor._pad_points_and_labels..r)Zaxis)max enumerater1r>Z concatenateZzerosrappend)rrrZexpected_nb_pointsZprocessed_input_pointsir+rrrr=s  z#SamProcessor._pad_points_and_labelsF)rcoordsrc Csx|\}}|jj||d\}}t|t}|r<|ddd}|d|||d<|d|||d<|rt|dd}|S)z~ Expects a numpy array of length 2 in the final dimension. Requires the original image size in (H, W) format. )r rB).r).r7r9)r Z_get_preprocess_shaperastypefloatZreshape) rrrGr.r4Zold_hZold_wZnew_hZnew_wrrrr(s z#SamProcessor._normalize_coordinatescCs|dk rNt|dr|}t|tr6t|dts>tddd|D}nd}|dk rt|drp|}t|trt|dtstddd|D}nd}|dk rt|dr|}t|trt|dtrt|ddtstd d d|D}nd}|||fS) a8 Check and preprocesses the 2D points, labels and bounding boxes. It checks if the input is valid and if they are, it converts the coordinates of the points and bounding boxes. If a user passes directly a `torch.Tensor`, it is converted to a `numpy.ndarray` and then to a `list`. Nrrz7Input points must be a list of list of floating points.cSsg|]}t|qSrr>r?)r*Z input_pointrrrr-sz=SamProcessor._check_and_preprocess_points..z-Input labels must be a list of list integers.cSsg|]}t|qSrrK)r*labelrrrr-sz>Input boxes must be a list of list of list of floating points.cSsg|]}t|tjqSr)r>r?rIZfloat32r5rrrr-s)rrtolist isinstancelist ValueError)rrrrrrrr s6         z)SamProcessor._check_and_preprocess_pointscCs|jj}tt|SN)r model_input_namesrOdictfromkeys)rZimage_processor_input_namesrrrrRszSamProcessor.model_input_namescOs|jj||SrQ)r post_process_masks)rargsr"rrrrUszSamProcessor.post_process_masks)NNNNN)NNNr%)F)NNN)__name__ __module__ __qualname____doc__ attributesZimage_processor_classrrrstrrrr$r!r=intr>Zndarrayr(r propertyrRrU __classcell__rrrrr #s@   . K  / r )rZcopyrtypingrrrr>Zprocessing_utilsrZtokenization_utils_baserutilsrr r r@Z tensorflowr8r rrrrs