+ "ݳj6D^RIt^RIt^RIHt^RIHtHt^RIt^RI t ^RI H t ^RI H t Rt RtRR ltR t!R R 4tRRR ] P$!R43RltRRR ] P$!R43RltRR] P$!R43RltRtRtRRltRRltR#)N)Thread)TupleUnion)Image)tqdmc\PP4'Ed=\PP^4P^8p\PP^4P^8pV'g\ P !R\^R7\;QJd8.R\PPR4R,4FNK 5M1!R\PPR4R,44pVR 8d0\ P !R\P R2\^R7VR 8;'gV'*pMRpR pRpWV3#) rzSFlash Attention is disabled as it requires a GPU with Ampere (8.0) CUDA capability.category stacklevelc38"TFp\V4xK R#5iN)int).0vs& `/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torchao/_models/sam2/utils/misc.py $get_sdpa_settings..sQ0P1A0Ps.NNzYou are using PyTorch zw without Flash Attention v2 support. Consider upgrading to PyTorch 2.2+ for Flash Attention v2 (which could be faster).TF)rr) torchcuda is_availableget_device_propertiesmajorwarningswarn UserWarningtuple __version__split)old_gpuuse_flash_attnpytorch_versionmath_kernel_ons rget_sdpa_settingsr&s zz  **2215;;a?99!<BBaG MMe$  %Q0A0A0G0G0LR0PQ%%Q0A0A0G0G0LR0PQQ V # MM():):(;BGK r'c\P!V4p\P!VP R4P W344pVP \P8Xd VR, pM\RVP RV 24h\P!V4P^^^4pVPwrVWFV3#)RGBo@zUnknown image dtype: z on ) ropennparrayconvertresizer8r- RuntimeErrorr from_numpypermutesize)img_path image_sizeimg_pilimg_npimg video_width video_heights&& r_load_img_as_tensorrj]sjj"G XXgooe,33Z4LM NF ||rxx%26<<.XJOPP   6 " * *1a 3C ' K k ))r'c6a]tRt^itoRtRtRtRtRtVt R#)AsyncVideoFrameLoaderzR A list of video frames to be load asynchronously without blocking session start. c<aVSnVSnVSnVSnVSnR.\ V4,SnRSnRSnRSn VSn SP^4V3Rlp\VRR7Sn SPP4R#)Nc<\\\SP44RR7FpSP V4K R# \ dpTSnRp?R#Rp?ii;i)frame loading (JPEG)descN)rrangelenimages __getitem__ Exception exception)neselfs r _load_frames4AsyncVideoFrameLoader.__init__.._load_framessP #eC $45  r'cVPe\R4VPhVPV,pVeV#\VPV,VP 4wr#pW0nW@nW P,pW P,pVP'gVPVPRR7pW PV&V#)NzFailure in frame loading threadT) non_blocking) rwr_rtrjrrdrirhrrrr,r)rzindexrgrirhs&& rru!AsyncVideoFrameLoader.__getitem__s >> %@At~~ Ukk%  ?J)< NN5 !4??* &;)& }} ||(((&&,,4&@C E r'c,\VP4#r )rsrt)rzs&r__len__AsyncVideoFrameLoader.__len__s4;;r') rrwrdrtrrrrrrirhN) __name__ __module__ __qualname____firstlineno____doc__rrur__static_attributes____classdictcell__) __classdict__s@rrlrlis$L*  r'rlFrc ~\V\4p\V\4pT;'d*\PP V4R,R9p V'g V 'd\ VVVVVVR7#V'd8\PPV4'd\VVVVVVVR7#\R4h)z Load the video frames from video_path. The frames are resized to image_size as in the model and are loaded to GPU if offload_video_to_cpu=False. This is used by the demo. ) video_pathrdrrrr)rrdrrrasync_loading_framesrz;Only MP4 video and JPEG folder are supported at this momentr>)z.mp4z.MP4) isinstancebytesstrospathsplitext!load_video_frames_from_video_fileisdir!load_video_frames_from_jpg_imagesNotImplementedError) rrdrrrrris_bytesis_str is_mp4_paths &&&&&&& rload_video_framesrs*e,H  C (FQQRWW--j9"=AQQK;0!!!5)    BGGMM*--0!!!5!5)  " I  r'cv\V\4'd)\PP V4'dTpM \ R4h\P !V4Uu.uF2p\PPV4R,R 9gK0VNK4 p pV PRR7\V 4p V ^8Xd\RV 24hV U u.uF"p \PPW{4NK$ p p \P!V\PR7R ,p\P!V\PR7R ,pV'd)\V VVVVV4p WP V P"3#\P$!V ^W\PR7p\'\)V RR74Fwpp\+VV4wW&ppK V'g4VP-V4pVP-V4pVP-V4pW,pW,pVXX3#uupiuup i) aD Load the video frames from a directory of JPEG files (".jpg" format). The frames are resized to image_size x image_size and are loaded to GPU if `offload_video_to_cpu` is `False` and to CPU if `offload_video_to_cpu` is `True`. You can load a frame asynchronously by setting `async_loading_frames` to `True`. akOnly JPEG frames are supported at this moment. For video files, you may use ffmpeg (https://ffmpeg.org/) to extract frames into a folder of JPEG files, such as ``` ffmpeg -i .mp4 -q:v 2 -start_number 0 /'%05d.jpg' ``` where `-q:v` generates high-quality JPEG frames and `-start_number 0` asks ffmpeg to start the JPEG file from 00000.jpg.c`\\PPV4^,4#)r)rrrr)ps&r3load_video_frames_from_jpg_images..s3rww'7'7':1'=#>r')keyzno images found in r8rorpr>)z.jpgz.jpegz.JPGz.JPEGNNNNN)rrrrrrlistdirrsortrsr_joinrtensorfloat32rlrirhzeros enumeraterrjr,)rrdrrrrr jpg_folderr frame_names num_frames frame_namer lazy_imagesrtrxrcrirhs&&&&&&& rrrs"*c""rww}}Z'@'@ ! <  J''A 77  A r "&H H ' >?[!JQ0 =>>HST *j5 IT||HEMM:=IHll7%--8GG+       44k6M6MMM [[Q emm TF i6L!MN 8/B8Z/X, <O >*;;~.**^, F F < ,,EUs"-H1H1(H6c^RIp\P!V\PR7R,p\P!V\PR7R,pVPP R4VP V4P4Pwrxp .p VP WVR7F%p V PV P^^^44K' \P!V ^R7P4R, p V'g4V PV4p VPV4pVPV4pW,p W,p WV3#)z(Load the video frames from a video file.Nrr)widthheightr;rYr)decordrrrbridge set_bridge VideoReadernextr?appendrarHfloatr,) rrdrrrrrrirhrJrtframes &&&&&& rrrs||HEMM:=IHll7%--8GG MMW%#)#5#5j#A#F#F#H#N#N Lq F##J#T emmAq!,-U[[Q ' - - /% 7F >*;;~.**^, F F  ,,r'c V^8gQR4hTp\V^8*4wr4V^8WA8*,p\P!VRV4pV# \d.p\P !T R2\ ^R7TpRp?T#Rp?ii;i)zQ A post processor to fill small holes in mask scores with area under `max_area`. zmax_area must be positiveg?a* Skipping the post-processing step due to the error above. You can still use SAM 2 and it's OK to ignore the error above, although some post-processing functionality may be limited (which doesn't affect the results in most cases; see https://github.com/facebookresearch/sam2/blob/main/INSTALL.md).r N)r0rrErvrrr)r/max_area input_masklabelsareasis_holerys&& rfill_holes_in_mask_scoresr9s a<444<J0; A:%"34{{7C. K   cN N!   K s6A B"A==BcVfYrCMB\P!VR,V.^R7p\P!VR,V.^R7pRVRV/#)zDAdd new points and labels to previous point inputs (add at the end). point_coordsr; point_labels)rcat)old_point_inputs new_points new_labelspointsrs&&& r concat_pointsrVsR#,^rs  3<J&4 *A A P# !<<'& Z# !<<'@-N# !<<' -@:<   r'