khGddlZddlZddlZddlZddlZddlmZddlmZm Z m Z m Z m Z m Z ddlZddlZddlmZddlmZddlmZ ddlZej0j3ej0j4eej8j:j<d sed Z ej@Z d1d Z#d e$fdZ%da&da' d2de(dejRde*de(de e e(efde ejRde e*de e(de e e(efd dfdZ+ddde*de*de(dd d!e e(e e e,e e,d"fe e,ffd e d#fd$Z-d%ejRd&e d#d'e,d(e*d ejRf d)Z. d3de(d*e e*efd+e e e*efde(d,e(d e ejRejRe e(efff d-Z/ddd e$fd.Z0ddd e e,fd/Z1d4de(de(d e e e,e e*ffd0Z2y#e!$rejDZ Y)wxYw#e$r ed ZY=wxYw)5N)Fraction)AnyDictListOptionalTupleUnion)_log_api_usage_once) _video_opt) _raise_video_deprecation_warning pict_typea+Your version of PyAV is too old for the necessary video operations in torchvision. If you are on Python 3.5, you will have to build from source (the conda-forge packages are not up-to-date). See https://github.com/mikeboers/PyAV#installation for instructions on how to install PyAV on your system. zPyAV is not installed, and is necessary for the video operations in torchvision. See https://github.com/mikeboers/PyAV#installation for instructions on how to install PyAV on your system. returnc8tttrtyN isinstanceav ExceptionP/var/www/teggl/fontify/venv/lib/python3.12/site-packages/torchvision/io/video.py_check_av_availabler,s"i !rc,ttt Srrrrr _av_availabler1s"i( ((r filename video_arrayfps video_codecoptions audio_array audio_fps audio_codec audio_optionsc (ttjjs-tjj st t ttj|tjjd}t|trtj|}tj |d5} | j#||} |j$d| _|j$d| _|d k7rd nd | _|xsi| _|2d d ddddddddd } | j#||} |xsi| _|j$d} | dkDrdnd}| j.j0dj2j4}tj6| |}tj|jdj9|}tj:j=|||}||_| jA|D]}| jC|| jAD]}| jC||D]R}tjDj=|d } d|_#| jA|D]}| jC|T| jAD]}| jC| d d d y #tH$rddl%m&}|jN|_#Y{wxYw#1swYy xYw)a~ [DEPRECATED] Writes a 4d tensor in [T, H, W, C] format in a video file. .. warning:: DEPRECATED: All the video decoding and encoding capabilities of torchvision are deprecated from version 0.22 and will be removed in version 0.24. We recommend that you migrate to `TorchCodec `__, where we'll consolidate the future decoding/encoding capabilities of PyTorch This function relies on PyAV (therefore, ultimately FFmpeg) to encode videos, you can get more fine-grained control by referring to the other options at your disposal within `the FFMpeg wiki `_. Args: filename (str): path where the video will be saved video_array (Tensor[T, H, W, C]): tensor containing the individual frames, as a uint8 tensor in [T, H, W, C] format fps (Number): video frames per second video_codec (str): the name of the video codec, i.e. "libx264", "h264", etc. options (Dict): dictionary containing options to be passed into the PyAV video stream. The list of options is codec-dependent and can all be found from `the FFMpeg wiki `_. audio_array (Tensor[C, N]): tensor containing the audio, where C is the number of channels and N is the number of samples audio_fps (Number): audio sample rate, typically 44100 or 48000 audio_codec (str): the name of the audio codec, i.e. "mp3", "aac", etc. audio_options (Dict): dictionary containing options to be passed into the PyAV audio stream. The list of options is codec-dependent and can all be found from `the FFMpeg wiki `_. Examples:: >>> # Creating libx264 video with CRF 17, for visually lossless footage: >>> >>> from torchvision.io import write_video >>> # 1000 frames of 100x100, 3-channel image. >>> vid = torch.randn(1000, 100, 100, 3, dtype = torch.uint8) >>> write_video("video.mp4", options = {"crf": "17"}) dtypeT)forcew)mode)rater r libx264rgbyuv420prgb24NzNONE) PictureType)(rtorchjit is_scripting is_tracingr write_videor as_tensoruint8numpyrfloatnproundropen add_streamshapewidthheightpix_fmtr"streamsaudior>namer)astype AudioFrame from_ndarray sample_rateencodemux VideoFramer TypeErrorav.video.framerAr@)rrr r!r"r#r$r%r& containerstreamaudio_format_dtypesa_stream num_channels audio_layoutaudio_sample_fmt format_dtypeframepacketimgrAs rrFrF:sj%& 99 ! ! #EII,@,@,BK(//+U[[AGGdGSK#uhhsm  $6" %%k%<"((+ #))!, &1\&Aw B  " # !++Ki+HH,2H &,,Q/L'3a'78VL(0066q9@@EE 88$78H$IJL//+6<<4<HOOP\]KMM..{CS\h.iE )E "//%0 & f% &#//+ & f% & &CMM..s7.CE 3"( !--. & f% & &mmo "F MM& ! "k6"6"X 36"-"2"2 3Y6"6"s2F0LK" AL" LLLLLr_zav.container.Container start_offset end_offsetpts_unitr`zav.stream.Stream stream_name.zav.frame.Framectdz attztdz k(rtj|dk(rkt t j |d|jz z}|tdk7rDt t j|d|jz z}ntjdi}d}d}|jdk(rv|jj} | r^d| vrZ| jd} | | d} t!j"d | } | t!j"d | } | | j%d d k(}|} t'| dz d } |rt'| |z d } |j)| dd|d } t-|j.di|D]2\}}|||j0<|j0|k\s%|r ||kr|dz }2nt3|Dcgc]#}|||j0cxkr|ksnn||%}}t5|d kDrL|d kDrG||vrC|Dcgc] }||ks | }}t5|d kDr t'|}|j7d |||S#t*$rgcYSwxYw#t*$rYwxYwcc}wcc}w)Nr secinfzBThe pts_unit 'pts' gives wrong results. Please use pts_unit 'sec'.TvideosDivXsDivX(\d+)Build(\d+)(\w)sDivX(\d+)b(\d+)(\w)prF) any_framebackwardr`r) _CALLED_TIMES_GC_COLLECTION_INTERVALgccollectintmathfloor time_baserJceilwarningswarntype codec_context extradatafindresearchgroupmaxseek FFmpegError enumeratedecodeptssortedleninsert)r_rjrkrlr`rmframes should_buffermax_buffer_sizerposdo seek_offset buffer_count_idxrgiresultpreceding_framesfirst_frame_ptss r_read_from_streamrsQM..2IA2MM 54::la&:J:J6J&KLM u %TYYzQ9I9I5I'JKLJ Z[ FMO {{g((22  I-..)C#$A 5q9AyII5q9} ! d 2 KkAoq)K+7; {ed6R L  $%5Y%5%5%D %DE KD% %F599 yyJ& \O%C A%L  "( _A<6!9==3^T^3^fQi _F _ 6{Q ""r start_ptsend_pts output_formatc ttjjs-tjj st t |j}|dvrtd|dddl m }|dk7rKtjj|std|tj ||||\}}}nt#| t%d }||krtd |d |i}g} g} tj&} t)j*|d 5} | j,j.r#| j,j.dj0} | j,j2r\t5| |||| j,j2dddi} | j,j2dj6} | t%| |d<| j,j.rOt5| |||| j,j.dddi} | j,j.dj8|d<ddd| Dcgc] }|j=j?"}}| Dcgc]}|j?}}|r)tj@tCjD|}n%tjFdtjH}|rtCjJ|d}tj@|}|dk(rVtMtOjP|d| z z}|t%d k7r$tMtOjR|d| z z}tU|| ||}n%tjFdtjV}|dk(r|jYdddd}|||fS#1swY|xYw#t:$rYwxYwcc}wcc}w)a[DEPRECATED] Reads a video from a file, returning both the video frames and the audio frames .. warning:: DEPRECATED: All the video decoding and encoding capabilities of torchvision are deprecated from version 0.22 and will be removed in version 0.24. We recommend that you migrate to `TorchCodec `__, where we'll consolidate the future decoding/encoding capabilities of PyTorch Args: filename (str): path to the video file. If using the pyav backend, this can be whatever ``av.open`` accepts. start_pts (int if pts_unit = 'pts', float / Fraction if pts_unit = 'sec', optional): The start presentation time of the video end_pts (int if pts_unit = 'pts', float / Fraction if pts_unit = 'sec', optional): The end presentation time pts_unit (str, optional): unit in which start_pts and end_pts values will be interpreted, either 'pts' or 'sec'. Defaults to 'pts'. output_format (str, optional): The format of the output video tensors. Can be either "THWC" (default) or "TCHW". Returns: vframes (Tensor[T, H, W, C] or Tensor[T, C, H, W]): the `T` video frames aframes (Tensor[K, L]): the audio frames, where `K` is the number of channels and `L` is the number of points info (Dict): metadata for the video and audio. Can contain the fields video_fps (float) and audio_fps (int) )THWCTCHWz5output_format should be either 'THWC' or 'TCHW', got .rget_video_backendpyavzFile not found: Nrpz7end_pts should be larger than start_pts, got start_pts=z and end_pts=ignoremetadata_errorsrr video_fpsrTr$)rr r rsr(r ro)r rrrsr )-rrBrCrDrEr read_videoupper ValueError torchvisionrospathexists RuntimeErrorr _read_videorrJdefault_timebaserrMrSrTr~rrr average_rater-rto_rgb to_ndarrayrGrKstackemptyrH concatenater{r|r}rrfloat32permute)rrrrlrrvframesrinfo video_framesraudio_timebaser_rrg vframes_list aframes_lists rrrs@%& 99 ! ! #EII,@,@,BJ'!'')M,,PQ^P__`abb-f$ww~~h'!1(<= =!+!7!7)WV^!_$ ?ElG Y I)Tabiajk   #44 8< H $$**%.%6%6%<%7I,J"KLG)'<GTGkk& >G//!Q1- GT !!o H H8   OEs7O/DN:O %O7O:O?O OOcZ|jdjj}|yd|vryy)NrFsLavcT)rSrr)r_rs r!_can_read_timestamps_from_packetsrs5!!!$22<`__, where we'll consolidate the future decoding/encoding capabilities of PyTorch Note that the function decodes the whole video frame-by-frame. Args: filename (str): path to the video file pts_unit (str, optional): unit in which timestamp values will be returned either 'pts' or 'sec'. Defaults to 'pts'. Returns: pts (List[int] if pts_unit = 'pts', List[Fraction] if pts_unit = 'sec'): presentation timestamps for each one of the frames in the video. video_fps (float, optional): the frame rate for the video rrrNrrz Failed decoding frames for file zFailed to open container for z; Caught error: ro)rrBrCrDrEr read_video_timestampsrrr _read_video_timestampsrrrMrSrrr~rrrrrJrRuntimeWarningsort) rrlrrrr_ video_streamvideo_time_baseemsgrs rrrst0%& 99 ! ! #EII,@,@,B12-f$008DDI C + WWXx 8 =I  &&(0066q9 "."8"8Q29=C",";";<  =HHJ5,/0qq?"00  >#QMM$DXJ"OPQ  = = +-hZ7GsK c>**+1sZ E%"rs+ :: 'H JJ))* 288>>,,k :   !nn  )t)  !(,*.!%!%.2v"v"v" v" v" d38n % v" %,,' v"v"#v"DcN+v" v"rM'MMM M  M c8E#uS#XS *I$JKKL M M` # \\ #)-.>)? #LO #Z_ # \\ #")*04 v"v"UH_%v"eE8O, -v" v"  v"  5<<tCH~ 56 v"r1IdO(@OT#YO8C8385cT\]bTcIcCd8y !jj !   Bs1A"G") G GG"GG"" G32G3