+ Ci0^RIt^RIHt^RIHt^RIt^RIt^RI t ^RI H t ^RI H t ^RIH t^RIHt^RIHtRR ltR R ltR R ltRtRt!RR]4tR#)N)deepcopy)List)validate)Image)map_coordinates) BaseSessioncV^8dQhR\PR\PR\R\P/#)imagematrix output_shapereturn)npndarraytuple)formats"P/var/www/html/photoedit/myenv/lib/python3.14/site-packages/rembg/sessions/sam.py __annotate__rs82&2& ::2&!zz2&9>2&ZZ2&c Vwr4\P!V.RO.4p\PPV4R,p\P!V4p\P!V4p\P !Wx4wrVR,V ,VR,V ,,VR ,,p VR ,V ,VR ,V ,,VR ,,p VP ^8Xd0\VP\P4W.^R^R7p M\P!W4VP^,3\PR7p \VP^,4FAp\VRRV3,P\P4W.^R^R7V RRV3&KC V PVP4#) a Apply affine transformation to an image (matching cv2.warpAffine behavior). cv2.warpAffine maps source coordinates to destination coordinates: dst(M @ [x, y, 1]^T) = src(x, y) So to fill dst(x', y'), we compute the inverse: src_coords = M^(-1) @ [x', y', 1]^T Args: image: Input image (H, W) or (H, W, C) matrix: 2x3 affine transformation matrix output_shape: (height, width) of output Returns: Transformed image Nr Nconstant)ordermodecvaldtypeNNNrr)rr)rr )rr )r r)r r )r r )rvstacklinalginvarangemeshgridndimrastypefloat64zerosshaperanger)r r r hwM_fullM_invcolsrowsx_coordsy_coordssrc_xsrc_yresultcs&&& r warp_affiner8sm( DAYY * +F IIMM& !" %E 99Qrc b.p\VP^,4Fxp.p\VP^,4FBpWV3,p\VVR,V^,V^,34pVPV4KD VPV4Kz \P !V4#)rr)r+r*r8rUrrV)masksrHtransform_matrix output_masksbatch batch_masksmask_idmasks&&& rtransform_masksrdmsLu{{1~& U[[^,G(D $q!=#34D   t $- K(' 88L !!rcha]tRt^toRtV3RlRltV3RlRlt]R4t]R4t Rt Vt R #) SamSessionz This class represents a session for the Sam model. Args: model_name (str): The name of the model. sess_opts (ort.SessionOptions): The session options. *args: Variable length argument list. **kwargs: Arbitrary keyword arguments. c:<V^8dQhRS[RS[P/#)r model_name sess_opts)strortSessionOptions)r __classdict__s"rrSamSession.__annotate__s%   %% rc WnVPP!V/VBp\P!\ V^,4VR7Vn\P!\ V^,4VR7VnR#)a Initialize a new SamSession with the given model name and session options. Args: model_name (str): The name of the model. sess_opts (ort.SessionOptions): The session options. *args: Variable length argument list. **kwargs: Arbitrary keyword arguments. ) sess_optionsN)rh __class__download_modelsrkInferenceSessionrjencoderdecoder)selfrhriargskwargspathss&&&*, r__init__SamSession.__init__sd %..??++ aM"  ++ aM"  rc6<V^8dQhRS[RS[S[,/#)r imgr)PILImager)rrms"rrrns'B4B4 B4 h B4rc VPRRRR^R\VP^, 4\VP^, 4./.4pRRRRRR RRR /RRR /RRRRRR /////p\ WER 7RpRpVP P 4^,PpVPR4p\P!V4p V PR,p V^,V P^,, p V^,V P^,, p \W4p \P!V ^^.^V ^..R O.4p\V VR,V^,V^,34p WP\P4/pVP P!RV4pV^,pRVRV RV/p\#V4wpp\P$!V\P!RR..4.^R7R!,p\P$!V\P!R".4.^R7R#,P\P4p\'VWv4P\P4p\P$!V\P(!^VP^,^3\PR7.^R7p\P*!VVP,4pVR$,P\P4p\P.!R%\PR7p\P.!^\PR7pRVRVRVRVRVR\P!V\PR7/pVP0P!RV4wpp\P2P5V4p\7VV V4p\P.!VP^,VP^,^3\P8R7pVR&,F p.R'OVVR8&K \:P<!V4PR4.#)(a5 Predict masks for an input image. This function takes an image as input and performs various preprocessing steps on the image. It then runs the image through an encoder to obtain an image embedding. The function also takes input labels and points as additional arguments. It concatenates the input points and labels with padding and transforms them. It creates an empty mask input and an indicator for no mask. The function then passes the image embedding, point coordinates, point labels, mask input, and has mask input to a decoder. The decoder generates masks based on the input and returns them as a list of images. Parameters: img (PILImage): The input image. *args: Additional arguments. **kwargs: Additional keyword arguments. Returns: List[PILImage]: A list of masks generated by the decoder. sam_promptrPrQrSrRrVitemsobject propertiesstringintegernumber)instanceschemaRGBrNimage_embeddingrHr^g)axisrimage_embeddings point_coords point_labels mask_inputhas_mask_input orig_im_sizeL)irr)Nrr)Nr)rrr)r r r)rrrr)rr)getr=widthheightrrt get_inputsnameconvertrrVr*minr8r'float32runr[ concatenaterNonesmatmulTr)rur"r#rduint8r fromarray)rvr}rwrxrWr target_size input_sizeencoder_input_namecv_imagerHscale_xscale_yr@r^encoder_inputsencoder_outputr embedding input_points input_labels onnx_coord onnx_labelonnx_mask_inputonnx_has_mask_inputdecoder_inputsr]_inv_transform_matrixrcms&&*, rpredictSamSession.predicts& GQSQ/SZZ!^1DE    G VX.fi0&(!3    &0  !\\446q9>>kk% 88C= r* Q-(.."33Q-(.."33G%881 E1     R ]JqM *   ; ))$?(+  ]  0 &6f%=" l^^\288c3ZL3I$JQRS  ^^\288RD>$BK  &  "*jFMM JJ ^^J,,Q/32::F   YYz+;+=+=> )00< ((#32::F hhq ;  J J / 1 BHHZrzzB  ll&&t^< q!!yy}}-=>}6JKxxQQ;288Lz""A+DSM#%--c233rc  VPRR4pVPRR4pV R2pV R2pV'd V R2pV R2p\P!R V 2R VVP!V/VBR R 7\P!R V 2R VVP!V/VBR R 7VR 8XEdt\P P \P PVP!V/VBR44'Eg \4p\^^4Fp\P!RV R2R RV R2VP!V/VBR R 7\P PVP!V/VBRV R24p VP\V R4P44\P!V 4K \\P PVP!V/VBR4R4;_uu_4p V PV4R R R 4\P PVP!V/VBV4\P PVP!V/VBV43# +'giLp;i)a Class method to download ONNX model files. This method is responsible for downloading two ONNX model files from specified URLs and saving them locally. The downloaded files are saved with the naming convention 'name_encoder.onnx' and 'name_decoder.onnx', where 'name' is the value returned by the 'name' method. Parameters: cls: The class object. *args: Variable length argument list. **kwargs: Arbitrary keyword arguments. Returns: tuple: A tuple containing the file paths of the downloaded encoder and decoder models. sam_modelsam_vit_b_01ec64 sam_quantFz .encoder.onnxz .decoder.onnxz.encoder.quant.onnxz.decoder.quant.onnxz>https://github.com/danielgatis/rembg/releases/download/v0.0.0/NT)fnamepath progressbarzsam_vit_h_4b8939.encoder.onnxz!sam_vit_h_4b8939.encoder_data.binz\https://github.com/danielgatis/rembg/releases/download/v0.0.0/sam_vit_h_4b8939.encoder_data.z.binzsam_vit_h_4b8939.encoder_data.rbwb)rpoochretrieve u2net_homeosrexistsjoin bytearrayr+extendopenreadremovewrite) clsrwrxrhquant fname_encoder fname_decodercontentifbinfps &*, rrrSamSession.download_models*sWZZ -?@  ;.%,m4 %,m4 )l*=>M)l*=>M L]O \ 00   L]O \ 00   ; ;BGGNN GGLL//1T E E  kG1a[rstruuyz:1#TB88 $ ww||NND3F34QCt<tD$/4467 $!  NND3F37  ! GGLL88- H GGLL88- H  s >I88 J c R#)z Class method to return a string value. This method returns the string value 'sam'. Parameters: cls: The class object. *args: Variable length argument list. **kwargs: Arbitrary keyword arguments. Returns: str: The string value 'sam'. sam)rrwrxs&*,rrSamSession.namexsr)rurtrhN) __name__ __module__ __qualname____firstlineno____doc__rzr classmethodrrr__static_attributes____classdictcell__)rms@rrfrfsL  8B4B4HK K Zrrf)rcopyrtypingrnumpyr onnxruntimerkr jsonschemarPILr PIL.Imager~ scipy.ndimagerbaserr8rCrNr[rdrfrrrrsN  ')2&j $"$HHr