U ʹj:= @s*ddlZddlZddlZddlZddlZddlmZddlmZm Z m Z m Z m Z m Z ddlZddlZddlmZddlmZz4ddlZejejjeejjjdsed ZWnek red ZYnXdd d d Zed ddZ da!da"d,e#ej$e%e#e e e#efe ej$e e%e e#e e e#efdd ddZ&de%e%e#de e#e e e'e e'dfe e'ffe ddddZ(ej$e de'e%ej$dddZ)d-e#e e%efe e e%efe#e#e ej$ej$e e#effd!d"d#Z*ded$d%d&Z+de e'd$d'd(Z,d.e#e#e e e'e e%fd)d*d+Z-dS)/N)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. )returncCstttrtdSN isinstanceav ExceptionrrS/var/www/html/bwave/python/venv/lib/python3.8/site-packages/torchvision/io/video.py_check_av_available's rcCs ttt Srrrrrr _av_available,sr libx264) filename video_arrayfps video_codecoptions audio_array audio_fps audio_codec audio_optionsrc  Cstjstjsttttj|tjd }t |t rJt |}tj|ddv} | j||d} |jd| _|jd| _|dkrdnd | _|pi| _|d k rtd d d d d d ddddd } | j||d} |pi| _|jd} | dkrdnd}| jjdjj}t | |}t| |}tjj|||d}||_| |D]}| |qH| D]}| |qb|D]8}tj!j|d d}d|_"| |D]}| |qqx| D]}| |qW5QRXd S)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 Z libx264rgbZyuv420pZrgb24NzgYSXd } zXt|jf|D]B\}}||j<|jkrV|r| |kr| d7} qVqqVWntjk rYnXfddtD}td kr*d kr*kr*fddD}t|d kr*t|}|d ||S)Nr secinfzBThe pts_unit 'pts' gives wrong results. Please use pts_unit 'sec'.TvideosDivXsDivX(\d+)Build(\d+)(\w)sDivX(\d+)b(\d+)(\w)prF)Z any_frameZbackwardrBcs2g|]*}|jkr"krnq|qSrpts.0irFframesrErr s  z%_read_from_stream..csg|]}|kr|qSrrrQ)rErrrVs) _CALLED_TIMES_GC_COLLECTION_INTERVALgcZcollectintmathfloor time_baser5ceilwarningswarntype codec_context extradatafindresearchgroupmaxseekrAVError enumeratedecoderPsortedleninsert)rArErFrGrBrHZ should_buffermax_buffer_sizercposdoZ seek_offsetZ buffer_countZ_idxrCresultZpreceding_framesZfirst_frame_ptsrrTr_read_from_streams^            "ru)aframes audio_frames ref_startref_endrc Cs||dj|dj}}|jd}||d|}d}|} ||krPt|||}||krht|||} |dd|| fS)Nrr )rPr9rZ) rvrwrxrystartendZ total_aframesZstep_per_aframeZs_idxZe_idxrrr_align_audio_framess r}rPTHWC)r start_ptsend_ptsrG output_formatrc Cstjstjstt|}|dkrt| |||| jjdddi} | jjdj} | dk r>t| |d<| jjrxt| |||| jjdddi} | jjdj|d<W5QRXWntjk rYnXdd| D}dd| D}|rtt|}ntj dtj!d}|rRt"|d}t|}|dkrBt#t$%|d| }|td krBt#t$&|d| }t'|| ||}ntj dtj(d}|dkr||)dddd}|||fS)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) )r~TCHWz5output_format should be either 'THWC' or 'TCHW', got .rget_video_backendzFile not found: pyavNrJz7end_pts should be larger than start_pts, got start_pts=z and end_pts=ignoreZmetadata_errorsrL video_fpsr=r cSsg|]}|qSr)Zto_rgb to_ndarrayrRrCrrrrVDszread_video..cSsg|] }|qSr)rrrrrrVEs)rr r rMr#r rI)r rrrMr )*r-r.r/r0r read_videoupper ValueError torchvisionrospathexists RuntimeErrorr Z _read_videorr5Zdefault_timebaserr8r<r=r]rLru average_rater'rjr2r6stackemptyr3Z concatenaterZr[r\r^r}Zfloat32Zpermute)rrrrGrrZvframesrvinfoZ video_framesrwZaudio_timebaserArZ vframes_listZ aframes_listrrrrs              r)rArcCs*|jdjj}|dkrdSd|kr&dSdS)NrFsLavcT)r<rbrc)rArcrrr!_can_read_timestamps_from_packets^s rcCs8t|rdd|jddDSdd|jddDSdS)NcSsg|]}|jdk r|jqSrrOrRxrrrrVjs z,_decode_video_timestamps..r)rLcSsg|]}|jdk r|jqSrrOrrrrrVls )rZdemuxrl)rArrr_decode_video_timestampsgsr)rrGrc s2tjstjsttddlm}|dkr>t ||St d}g}zvt j |dd^}|j jr|j jd}|jz t|}Wn&t jk rtd|YnXt|j}W5QRXWnBt jk r}z d|d |}t|tW5d}~XYnX||d kr*fd d |D}||fS) a  List the video frames timestamps. 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: rIcsg|] }|qSrrrZvideo_time_baserrrVsz)read_video_timestamps..)r-r.r/r0r read_video_timestampsrrr Z_read_video_timestampsrrr8r<rLr]rrjr_r`r5rRuntimeWarningsort) rrGrrrPrAZ video_streamemsgrrrros2      r)rNNNNN)rNrPr~)rP).rYr[rrer_ fractionsrtypingrrrrrrr4r6r-utilsr r rloggingZ set_levelERRORhasattrrLrCr@ ImportErrorrboolrrWrXstrZTensorr5r1rZrur}rrrrrrrrs       Y$ Q   r