jQ%ddlZddlZddlmZddlmZmZmZmZm Z ddl Z ddl m Z  e ddZ n#eef$rdZ YnwxYwe Zedd ZGd d ZGd d ZdeeefddfdZde jde jde jde jde jde jdefdZde jde jdeeefde jfdZdddddddeddddef deded ed!ed"ed#ed$ed%eeefd&ed'ed(ed)edeeefd*edee je jeffd+Zdedeeeeeeffd,Zdedefd-Z dded9edee eeeefeeffd;Z#dS)?N)Fraction)DictListOptionalTupleUnion) _load_library video_readerTFc4eZdZeedZddgZdededdfdZdS)Timebase numerator denominatorrrreturnNc"||_||_dS)Nr)selfrrs JC:\Projects\Historical\venv\Lib\site-packages\torchvision/io/_video_opt.py__init__zTimebase.__init__s #&)__name__ __module__ __qualname__int__annotations__ __slots__rrrrrs[$'<+>'?'?Yq\EVEVEXEXAYAYZZQ<$$&&y|/@/@/B/B)C)CC ??  q !DN"' (8(8"9"9H"DD  zz||atyy{{++1&s9Q<+<+<+>+>'?'?Yq\EVEVEXEXAYAYZZQ<$$&&y|/@/@/B/B)C)CC ??  q !DN"' (8(8"9"9H"DD a!&|'8'8':':!;!; Kraframes aframe_ptsaudio_pts_rangecn|d|d}}|d}t||z dzt|z }d}|}||dkrt|d|z |z }|ddkr'||dkrt|d|z |z }|||ddfS)Nrr )sizer+r) r<r=r>startend num_samplesstep_per_aframes_idxe_idxs r_align_audio_framesrHmsA 23E,,q//KC%K!O,,u[/A/AAO E E q!!!_Q'%/?BCCqRC/!*<$<$<_Q'#-@AA 5;> ""r?rr@filenameseek_frame_marginread_video_stream video_width video_heightvideo_min_dimensionvideo_max_dimensionvideo_pts_ranger"read_audio_stream audio_samplesaudio_channelsr&ct|t| tjj||d||||||d|d|j|j| | | | d| d| j| j}|\ }}}}}}}}}}t||||||}|dkrt||| }|||fS)ab Reads a video from a file, returning both the video frames and the audio frames Args: filename (str): path to the video file seek_frame_margin (double, optional): seeking frame in the stream is imprecise. Thus, when video_start_pts is specified, we seek the pts earlier by seek_frame_margin seconds read_video_stream (int, optional): whether read video stream. If yes, set to 1. Otherwise, 0 video_width/video_height/video_min_dimension/video_max_dimension (int): together decide the size of decoded frames: - When video_width = 0, video_height = 0, video_min_dimension = 0, and video_max_dimension = 0, keep the original frame resolution - When video_width = 0, video_height = 0, video_min_dimension != 0, and video_max_dimension = 0, keep the aspect ratio and resize the frame so that shorter edge size is video_min_dimension - When video_width = 0, video_height = 0, video_min_dimension = 0, and video_max_dimension != 0, keep the aspect ratio and resize the frame so that longer edge size is video_max_dimension - When video_width = 0, video_height = 0, video_min_dimension != 0, and video_max_dimension != 0, resize the frame so that shorter edge size is video_min_dimension, and longer edge size is video_max_dimension. The aspect ratio may not be preserved - When video_width = 0, video_height != 0, video_min_dimension = 0, and video_max_dimension = 0, keep the aspect ratio and resize the frame so that frame video_height is $video_height - When video_width != 0, video_height == 0, video_min_dimension = 0, and video_max_dimension = 0, keep the aspect ratio and resize the frame so that frame video_width is $video_width - When video_width != 0, video_height != 0, video_min_dimension = 0, and video_max_dimension = 0, resize the frame so that frame video_width and video_height are set to $video_width and $video_height, respectively video_pts_range (list(int), optional): the start and end presentation timestamp of video stream video_timebase (Fraction, optional): a Fraction rational number which denotes timebase in video stream read_audio_stream (int, optional): whether read audio stream. If yes, set to 1. Otherwise, 0 audio_samples (int, optional): audio sampling rate audio_channels (int optional): audio channels audio_pts_range (list(int), optional): the start and end presentation timestamp of audio stream audio_timebase (Fraction, optional): a Fraction rational number which denotes time base in audio stream Returns vframes (Tensor[T, H, W, C]): the `T` video frames aframes (Tensor[L, K]): the audio frames, where `L` is the number of points and `K` is the number of audio_channels info (Dict): metadata for the video and audio. Can contain the fields video_fps (float) and audio_fps (int) rr ) r/torchopsr read_video_from_filerrr;r7rH)rKrLrMrNrOrPrQrRr"rSrTrUr>r&resultvframes _vframe_ptsr0r1r2r<r=r3r4r5infos r_read_video_from_filer^|s@/"""/""" Y # 8 8  " "'F*qwmG[)T9gz9Vbdm iy)\9 U UD}}%gz?KK GT !!rcRtjj|dddddddddddddddddd}|\ }}}}}}}} } } t |||| | | } |}|}||| fS)z Decode all video- and audio frames in the video. Only pts (presentation timestamp) is returned. The actual frame pixel data is not copied. Thus, it is much faster than read_video(...) rr r@)rWrXr rYr;numpytolist) rKrZ_vframes vframe_ptsr0r1r2_aframesr=r3r4r5r]s r _read_video_timestamps_from_fileres Y # 8 8                  'F*rxnHj)T9h IWcen iy)\9 U UD!!##**,,J!!##**,,J z4 ''rctjj|}|\}}}}}}t ||||||}|S)zO Probe a video file and return VideoMetaData with info about the video )rWrXr probe_video_from_filer;) rKrZr0r1r2r3r4r5r]s r_probe_video_from_filerhsJY # 9 9( C CFEKBItY < iy)\9 U UD Kr video_datavideo_timebase_numeratorvideo_timebase_denominatoraudio_timebase_numeratoraudio_timebase_denominatorcDt|t| t|tjsat j5t jddtj|tj}dddn #1swxYwYtj j ||d||||||d|d|| | | | | d| d||}|\ }}}}}}}}}}| dkrt||| }||fS)a Reads a video from memory, returning both the video frames as the audio frames This function is torchscriptable. Args: video_data (data type could be 1) torch.Tensor, dtype=torch.int8 or 2) python bytes): compressed video content stored in either 1) torch.Tensor 2) python bytes seek_frame_margin (double, optional): seeking frame in the stream is imprecise. Thus, when video_start_pts is specified, we seek the pts earlier by seek_frame_margin seconds read_video_stream (int, optional): whether read video stream. If yes, set to 1. Otherwise, 0 video_width/video_height/video_min_dimension/video_max_dimension (int): together decide the size of decoded frames: - When video_width = 0, video_height = 0, video_min_dimension = 0, and video_max_dimension = 0, keep the original frame resolution - When video_width = 0, video_height = 0, video_min_dimension != 0, and video_max_dimension = 0, keep the aspect ratio and resize the frame so that shorter edge size is video_min_dimension - When video_width = 0, video_height = 0, video_min_dimension = 0, and video_max_dimension != 0, keep the aspect ratio and resize the frame so that longer edge size is video_max_dimension - When video_width = 0, video_height = 0, video_min_dimension != 0, and video_max_dimension != 0, resize the frame so that shorter edge size is video_min_dimension, and longer edge size is video_max_dimension. The aspect ratio may not be preserved - When video_width = 0, video_height != 0, video_min_dimension = 0, and video_max_dimension = 0, keep the aspect ratio and resize the frame so that frame video_height is $video_height - When video_width != 0, video_height == 0, video_min_dimension = 0, and video_max_dimension = 0, keep the aspect ratio and resize the frame so that frame video_width is $video_width - When video_width != 0, video_height != 0, video_min_dimension = 0, and video_max_dimension = 0, resize the frame so that frame video_width and video_height are set to $video_width and $video_height, respectively video_pts_range (list(int), optional): the start and end presentation timestamp of video stream video_timebase_numerator / video_timebase_denominator (float, optional): a rational number which denotes timebase in video stream read_audio_stream (int, optional): whether read audio stream. If yes, set to 1. Otherwise, 0 audio_samples (int, optional): audio sampling rate audio_channels (int optional): audio audio_channels audio_pts_range (list(int), optional): the start and end presentation timestamp of audio stream audio_timebase_numerator / audio_timebase_denominator (float, optional): a rational number which denotes time base in audio stream Returns: vframes (Tensor[T, H, W, C]): the `T` video frames aframes (Tensor[L, K]): the audio frames, where `L` is the number of points and `K` is the number of channels ignore The given buffer is not writablemessagedtypeNrr )r/ isinstancerWTensorwarningscatch_warningsfilterwarnings frombufferuint8rXr read_video_from_memoryr7rH)rirLrMrNrOrPrQrRrjrkrSrTrUr>rlrmrZr[r\r0r1r2r<r=r3r4r5s r_read_video_from_memoryr} sJ/"""/""" j%, / /I  $ & & I I  #H6X Y Y Y Y)*EKHHHJ I I I I I I I I I I I I I I I Y # : :  " "'F,qwmG[)T9gz9Vbdm}}%gz?KK G s 7BBBcHt|tjsatj5tjddtj|tj}dddn #1swxYwYtjj |dddddddddddddddddd}|\ }}}}}}}} } } t|||| | | } | }| }||| fS) z Decode all frames in the video. Only pts (presentation timestamp) is returned. The actual frame pixel data is not copied. Thus, read_video_timestamps(...) is much faster than read_video(...) rorprqrsNrr r@)rurWrvrwrxryrzr{rXr r|r;r`ra) rirZrbrcr0r1r2rdr=r3r4r5r]s r"_read_video_timestamps_from_memoryrvs j%, / /I  $ & & I I  #H6X Y Y Y Y)*EKHHHJ I I I I I I I I I I I I I I IY # : :                  'F*rxnHj)T9h IWcen iy)\9 U UD!!##**,,J!!##**,,J z4 ''7A11A58A5c~t|tjsatj5tjddtj|tj}dddn #1swxYwYtjj |}|\}}}}}}t||||||}|S)zy Probe a video in memory and return VideoMetaData with info about the video This function is torchscriptable rorprqrsN) rurWrvrwrxryrzr{rXr probe_video_from_memoryr;) rirZr0r1r2r3r4r5r]s r_probe_video_from_memoryrs j%, / /I  $ & & I I  #H6X Y Y Y Y)*EKHHHJ I I I I I I I I I I I I I I IY # ; ;J G GFEKBItY < iy)\9 U UD Krpts start_ptsend_ptspts_unitc tddkrtjdt|}|j}|j}fd}d}t } |r/t|jj |jj } || }d} t } |r/t|j j |j j } || } t|d|| d| | \} } }i}|r |j |d<|r |j|d <| | |fS) NinfrmThe pts_unit 'pts' gives wrong results and will be removed in a follow-up version. Please use pts_unit 'sec'.c }}dkrattjd|z z}|tdkr'ttjd|z z}|tdkrd}||fS)Nsecr rr@)rmathfloorr+ceil) time_base start_offset end_offsetrrrs rget_ptsz_read_video..get_ptss  u  tz)q9}*EFFGGLU5\\)) 7a)m+D!E!EFF u % %JZ''rrJT)rMrRr"rSr>r&r$ audio_fps)r+rwwarnrhr!r%default_timebaserr"rrr&r^r$r()rKrrrr]r!r%rrRr"r>r&r[r<_infos ``` r _read_videorsb ,,5  >   "( + +DII ( ( ( ( ( ( (O%N2!$"5"?ATA`aa!'.11O%N2!$"5"?ATA`aa!'.112'%'%GWd E,!^k4!3k GU ""rc|dkrtjdt|\}}}|dkr2t|jj|jjfd|D}|jr|jnd}||fS)Nrrrcg|]}|zSrr).0xvideo_time_bases r z*_read_video_timestamps..s000qq?"000r) rwrrerr"rrr!r$)rKrr_r]r$rs @r_read_video_timestampsrs5  >   4H==LCD5"4#6#@$BUBabb0000C000"&.:dI  >r)rIr rrrrrJrr r rrrJrr )rNr)r)$rrw fractionsrtypingrrrrrrW extensionr _HAS_CPU_VIDEO_DECODER ImportErrorOSError_HAS_VIDEO_OPTrrr rr/rvr;rHstrr+r*r^rerhr}rrrrrrrrs 55555555555555 %%%%%%#M.!!!!W###"#(8Aq>> ' ' ' ' ' ' ' '%%%%%%%%B U38_     | ,||  ,  | @ # \ #',| #FKCQTHo # \ # # # #" $"  './"'./]"]"]"]"]" ]"  ]"  ]"]"38_]"]"]"]"]"38_]"]" 5<} 45]"]"]"]"@ (s (uT#YS S`=`7a ( ( ( (FS] $  '.$%&''.$%&'!jj jjj j  j  jj38_j"j!$jjjj38_j"j !$!j" 5< %&#jjjjZ'( '( 49d3i ./'('('('(T ()*04 :#:#:#UH_%:#eE8O, -:# :#  5<tCJ'7 78 :#:#:#:#|$)  5cDN* +Xe_ <=s 6 AA