,}h:=ddlZddlZddlZddlZddlZddlmZddlmZm Z m Z m Z m Z m Z ddlZddlZddlmZddlmZ ddlZej,j/ej,j0eej4j6j8dsed Zd0d Zd efd Z da!da" d1de#dejHde%de#de e e#efde ejHde e%de e#de e e#efd dfdZ&ddde%de%de#ddd e e#e e e'e e'd!fe e'ffd e d"fd#Z(d$ejHd%e d"d&e'd'e%d ejHf d(Z) d2de#d)e e%efd*e e e%efde#d+e#d e ejHejHe e#efff d,Z*ddd efd-Z+ddd e e'fd.Z,d3de#de#d e e e'e e%ffd/Z-y#e$r ed ZY%wxYw)4N)Fraction)AnyDictListOptionalTupleUnion)_log_api_usage_once) _video_opt 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 ExceptionS/var/www/html/test/engine/venv/lib/python3.12/site-packages/torchvision/io/video.py_check_av_availabler's"i !rc,ttt Srrrrr _av_availabler,s"i( ((r filename video_arrayfps video_codecoptions audio_array audio_fps audio_codec audio_optionsc tjjs-tjjst t t tj|tjj}t|trtj|}tj|d5} | j!||} |j"d| _|j"d| _|dk7rdnd | _|xsi| _|0d d d d d d ddddd } | j!||} |xsi| _|j"d} | dkDrdnd}| j,j.dj0j2}tj4| |}tj|jj7|}tj8j;|||}||_| j?|D]}| jA|| j?D]}| jA||D]Q}tjBj;|d }d|_"| j?|D]}| jA|S| j?D]}| jA| d d d y #1swYy xYw)a| Writes a 4d tensor in [T, H, W, C] format in a video file 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 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 dtypew)mode)rater r libx264rgbyuv420prgb24NzEElSKMM..{CS\h.iE )E "//%0 & f% &#//+ & f% & &CMM..s7.CE$EO --. & f% & &mmo "F MM& ! "_0"0"0"s HKKrZzav.container.Container start_offset end_offsetpts_unitr[zav.stream.Stream stream_name.zav.frame.FramecBtdz 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/|j0di|D]2\}}|||j2<|j2|k\s%|r ||kr|dz }2nt5|Dcgc]#}|||j2cxkr|ksnn||%}}t7|d kDrL|d kDrG||vrC|Dcgc] }||ks | }}t7|d kDr t'|}|j9d |||S#t*j,$rgcYSwxYw#t*j,$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_baserGceilwarningswarntype codec_context extradatafindresearchgroupmaxseekrAVError enumeratedecodeptssortedleninsert)rZrerfrgr[rhframes should_buffermax_buffer_sizerposdo seek_offset buffer_count_idxrbiresultpreceding_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{QI>J; J" J-J#I;:I;>JJaframes audio_frames ref_startref_endc|dj|dj}}|jd}||z dz|z }d}|} ||krt||z |z }||kDrt||z |z } |dd|| fS)Nrr )rrLrv) rrrrstartend total_aframesstep_per_aframes_idxe_idxs r_align_audio_framesrsa$$l2&6&:&:3EMM!$MU{Q-7O E E yY&/9: W}Ws]o56 1eEk> ""r start_ptsend_pts output_formatc tjjs-tjjst t |j }|dvrtd|dddlm }tjj|std||dk7rtj||||\}}}nt!| t#d }||krtd |d |i}g} g} tj$} t'j(|d 5} | j*j,r#| j*j,dj.} | j*j0r\t3| |||| j*j0dddi} | j*j0dj4} | t#| |d<| j*j,rOt3| |||| j*j,dddi} | j*j,dj6|d<ddd| Dcgc] }|j;j="}}| Dcgc]}|j=}}|r)tj>tAjB|}n%tjDdtjF}|rtAjH|d}tj>|}|dk(rVtKtMjN|d| z z}|t#d k7r$tKtMjP|d| z z}tS|| ||}n%tjDdtjT}|dk(r|jWdddd}|||fS#1swY|xYw#t&j8$rYwxYwcc}wcc}w)a Reads a video from a file, returning both the video frames and the audio frames Args: filename (str): path to the video file 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_backendzFile not found: pyavNrkz7end_pts should be larger than start_pts, got start_pts=z and end_pts=ignoremetadata_errorsrm video_fpsrQr#)rr r rnr'r rj)r rrrnr ),r?r@rArBr read_videoupper ValueError torchvisionrospathexists RuntimeErrorr _read_videorrGdefault_timebaserrJrPrQryrmr average_rater+rto_rgb to_ndarrayrDrHstackemptyrE concatenatervrwrxrzrfloat32permute)rrrrgrrvframesrinfo video_framesraudio_timebaserZrrb vframes_list aframes_lists rrrs}2 99 ! ! #EII,@,@,BJ'!'')M,,PQ^P__`abb- 77>>( #-hZ899f$!+!7!7)WV^!_$ ?ElG Y I)Tabiajk   #44 8< H $$**%.%6%6%<%7I,J"KLG)'<GTGkk& >G//!Q1- GT !!o H H8zz   OEs7N=%DN06N=%O-O0N:5N==OOcZ|jdjj}|yd|vryy)NrFsLavcT)rPr~r)rZrs r!_can_read_timestamps_from_packetsr^s5!!!$22<zzQMM$DXJ"OPQ  = = ::+-hZ7GsK c>**+1sZE%rs :: ' JJ))* 288>>,,k :   & )t)  !(,*.!%!%.2U"U"U" U" U" d38n % U" %,,' U"U"#U"DcN+U" U"pM'MMM M  M c8E#uS#XS *I$JKKL M M` # \\ #)-.>)? #LO #Z_ # \\ #")*04 o"o"UH_%o"eE8O, -o" o"  o"  5<<tCH~ 56 o"d1IdO(@OT#YO0C0305cT\]bTcIcCd0e    BsA"F77 GG