fTh2 SrSSKJr SSKJrJrJr SSKrSSK J r SSK J r J r Jr SSKJrJrJrJr SS KJrJr SS KJr \"5(aSSKr\"5(aSSKr"S S \ 5r"S S\ SS9r"SS\5rS/rg)z Processor class for SAM. )deepcopy)ListOptionalUnionN) ImageInput) ImagesKwargsProcessingKwargsProcessorMixin) AudioInput BatchEncodingPreTokenizedInput TextInput)is_tf_availableis_torch_available) VideoInputc\rSrSr%\\\S'\\\\\S'\\\\ \S'\\\\\\S'\\ \S'Sr g) SamImagesKwargs&segmentation_maps input_points input_labels input_boxespoint_pad_valueN) __name__ __module__ __qualname____firstlineno__rr__annotations__rfloatint__static_attributes__r^/var/www/auris/envauris/lib/python3.13/site-packages/transformers/models/sam/processing_sam.pyrr&sW ++4U ,--4S ?++$tDK0122c]"r$rc,\rSrSr%\\S'SSS00rSrg)SamProcessorKwargs. images_kwargsrrN)rrrrrr _defaultsr#rr$r%r'r'.s"" s Ir$r'F)totalc $^\rSrSrSrS/rSr/SQrU4SjrSSSSS.S \ \ S \ \ \ \ \\ \\ 4S \ \S \ \S \4 SjjjrSSjrSrSS\S\R.S \R.4SjjrSSjr\S5rSrSrU=r$) SamProcessor7a 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_processorSamImageProcessor)rrrrc`>[TU]U5 URRSUlg)N longest_edge)super__init__r0size target_size)selfr0 __class__s r%r5SamProcessor.__init__Ns) )//44^Dr$N)textaudiovideoimagesr;r<r=returnc UR"[4S00UDUR"U6D6nUSRSS5nUSRSS5n USRSS5n USRSS5n UR"U40USD6n U Sn [ U S 5(aU R 5n URUU U S 9upn URU U UU U US RS 5U S 9n U $)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. tokenizer_init_kwargsr)rNrrroriginal_sizesnumpy)rrr common_kwargsreturn_tensors)rrrrEr) _merge_kwargsr''prepare_and_validate_optional_call_argspopr0hasattrrC_check_and_preprocess_points_normalize_and_convertget)r8r>r;r<r=argskwargs output_kwargsrrrrencoding_image_processorrBs r%__call__SamProcessor.__call__RsQ&**  "$  ::DA  %_599.$O $_599.$O #O488M '8<<=NPTU#'#7#7 $ O,$  22BC >7 + ++113N262S2S%%#3T3 / K $(#>#> $ %%#(9==>NO+$?$  ('r$c @^Tb[U5[T5:wa0TVs/sH!oRURXS5PM# snmO;[TU5VV s/sH!upURURX5PM# sn nm[ U4SjT55(dUbUR TXG5umn[ R"T5mUb[ R"U5nUb[U5[U5:wa/UV s/sH!n URURXSSS9PM# nn O8[XR5V V s/sH upURURXSS9PM" nn n [ R"U5nUbUS:XaC[R"U5n[UR5S:waURS5OUnONUS:XaH[R"U5n[UR5S:wa[R"US5OUnURS U05 TbUS:XaC[R"T5m[TR5S :waTRS5OTmONUS:XaH[R"T5m[TR5S :wa[R"TS5OTmURS T05 UbUS:XaC[R"U5n[UR5S:waURS5OUnONUS:XaH[R"U5n[UR5S:wa[R"US5OUnURS U05 U$s snfs sn nfs sn fs sn n f) Nrc3Z># UH oRTSR:Hv M" g7f)rN)shape).0pointrs r% 6SamProcessor._normalize_and_convert..s"V{{l1o&;&;;s(+T)is_bounding_boxptrtfrrr)len_normalize_coordinatesr7zipall_pad_points_and_labelsnparraytorch from_numpyrU unsqueezer]convert_to_tensor expand_dimsupdate) r8rPrBrrrrErrW original_sizeboxs ` r%rK#SamProcessor._normalize_and_convertsj  #>"c,&77iu iu`e//0@0@%XYIZ[iu  14L.0Q 0Q,//0@0@%W0Q  VVVV+151L1L$l2.L,88L1L  #88L1L  ">"c+&66 +*//0@0@#VWGXjn/o* /2+.N.N*//0@0@#fj/k.N((;/K  "%#..{; :=k>O>O:PTU:Uk33A6[f 4' 22;? @CKDUDU@VZ[@[bnn[!dp $ + +^\,J K  #%$// = dp $ + +^\,J K''A  $ s(N (N(N'Nc [UVs/sHoDRSPM sn5n/n[U5HuptURSU:wa[[R"U[R "XTRS- S45U-/SS9n[R "X'U/5X''UR U5 M UnX4$s snf)zX The method pads the 2D points and labels to the maximum number of points in the batch. r)axis)maxrU enumeraterd concatenatezerosappend)r8rrrrWexpected_nb_pointsprocessed_input_pointsis r%rc#SamProcessor._pad_points_and_labelss!l!KlU++a.l!KL!#!,/HA{{1~!33BHH&8;;q>&I1%MNQ``ahi#%))LOo=N"O  " ) )% 0 0. ))"LsCr7coordscUupVURRX1S9upx[U5R[5nU(aUR SSS5nUSX- -US'USXu- -US'U(aUR SS5nU$)zn Expects a numpy array of length 2 in the final dimension. Requires the original image size in (H, W) format. )r3rp).r).r\r^)r0_get_preprocess_shaperastyper!reshape) r8r7r{rlrZold_hold_wnew_hnew_ws r%r`#SamProcessor._normalize_coordinatess % ++AA-Aj &!((/ ^^B1-F5=9v5=9v ^^B*F r$c Ub[US5(aUR5R5n[U[5(a[US[5(d [ S5eUVs/sHn[ R"U5PM nnOSnUb[US5(aUR5R5n[U[5(a[US[5(d [ S5eUVs/sHn[ R"U5PM nnOSnUb[US5(aUR5R5n[U[5(a3[US[5(a[USS[5(d [ S5eUVs/sH6n[ R"U5R[ R5PM8 nnOSnXU4$s snfs snfs snf)a 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`. NrCrz7Input points must be a list of list of floating points.z-Input labels must be a list of list integers.z>Input boxes must be a list of list of list of floating points.) rIrCtolist isinstancelist ValueErrorrdrerfloat32)r8rrr input_pointlabelrms r%rJ)SamProcessor._check_and_preprocess_pointss  #|W--+113::< lD11LQROUY9Z9Z !Z[[EQR\kBHH[1\LRLL  #|W--+113::< lD11LQROUY9Z9Z !PQQ9EFBHHUOLFLL  "{G,,)//188: {D11!+a.$77!+a."3T:: !abbGRS{288C=// ;{KSKK;669SGTs/ G6 G;0=HcjURRn[[R U55$N)r0model_input_namesrdictfromkeys)r8image_processor_input_namess r%rSamProcessor.model_input_names/s)&*&:&:&L&L#DMM"=>??r$c:URR"U0UD6$r)r0post_process_masks)r8rMrNs r%rSamProcessor.post_process_masks4s##66GGGr$)r7r)NNNr[r*)F)NNN)rrrr__doc__ attributesimage_processor_classoptional_call_argsr5rrrrrrr rr rQrKrcr"rdndarrayr`rJpropertyrrr# __classcell__)r9s@r%r.r.7s $$J/E (,9(im&*&*9($9(uY(94 ?DQbLccde9( #9( #9( 9(~L(\*"TY(*  . -7^@@HHr$r.) rcopyrtypingrrrrCrd image_utilsrprocessing_utilsr r r tokenization_utils_baser r rrutilsrr video_utilsrrf tensorflowr]rr'r.__all__rr$r%rs}((%NN^^8%#l#)~H>~HB  r$