%`ijddlZddlZejddZdZdZd dZdZd Zdd Z d Z dS)NzThe 'torchvision.transforms._functional_video' module is deprecated since 0.12 and will be removed in the future. Please use the 'torchvision.transforms.functional' module instead.ctj|stdt|z|dks$t d|zdS)Nzclip should be Tensor. Got %szclip should be 4D. Got %dDT)torch is_tensor TypeErrortype ndimension ValueErrordimclips |/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/torchvision/transforms/_functional_video.py_is_tensor_video_clipr se ?4 F7$t**DEEE ??   ! !5 BCCC 4ct|dkrtd|d|||z|||zfS)z[ Args: clip (torch.tensor): Video clip to be cropped. Size is (C, T, H, W) rzclip should be a 4D tensor.)lensizer )r ijhws rcroprsP  499;;15666 QQYAE ) **rct|dkrtd|tjj|||dS)Nz9target size should be tuple (height, width), instead got F)rmode align_corners)rr rnn functional interpolate)r target_sizeinterpolation_modes rresizer" sQ ;1bU`bbccc 8  * *4kHZjo * p pprbilinearct|stdt|||||}t|||}|S)a Do spatial cropping and resizing to the video clip Args: clip (torch.tensor): Video clip to be cropped. Size is (C, T, H, W) i (int): i in (i,j) i.e coordinates of the upper left corner. j (int): j in (i,j) i.e coordinates of the upper left corner. h (int): Height of the cropped region. w (int): Width of the cropped region. size (tuple(int, int)): height and width of resized clip Returns: clip (torch.tensor): Resized and cropped clip. Size is (C, T, H, W) clip should be a 4D torch.tensor)rr rr")r rrrrrr!s r resized_cropr&&sO ! & &=;<<< aAq ! !D $0 1 1D Krct|std|d|d}}|\}}||ks||krtdtt ||z dz }tt ||z dz }t |||||S)Nr%z2height and width must be no smaller than crop_sizeg@)rr rintroundr)r crop_sizerrthtwrrs r center_cropr/:s  & &=;<<< 99R==$))B--qA FB2vvRMNNN E1r6S. ! !""A E1r6S. ! !""A aB # ##rct||jtjks$t dt |jz|dddddz S)a Convert tensor data type from uint8 to float, divide value by 255.0 and permute the dimensions of clip tensor Args: clip (torch.tensor, dtype=torch.uint8): Size is (T, H, W, C) Return: clip (torch.tensor, dtype=torch.float): Size is (C, T, H, W) z/clip tensor should have data type uint8. Got %srrgo@)rdtyperuint8rstrfloatpermuter s r to_tensorr8Gsf$ : $ $ICPTPZOO[\\\ ::<<  1a + +e 33rFcrt|std|s|}tj||j|j}tj||j|j}||dddddf|dddddf|S)a Args: clip (torch.tensor): Video clip to be normalized. Size is (C, T, H, W) mean (tuple): pixel RGB mean. Size is (3) std (tuple): pixel standard deviation. Size is (3) Returns: normalized clip (torch.tensor): Size is (C, T, H, W) r%)r3deviceN) rr cloner as_tensorr3r:sub_div_)r meanstdinplaces r normalizerBVs ! & &=;<<< zz|| ?4tz$+ F F FD /#TZ D D DCIId111dD$&'((--c!!!T42E.FGGG Krcht|std|dS)z Args: clip (torch.tensor): Video clip to be normalized. Size is (C, T, H, W) Returns: flipped clip (torch.tensor): Size is (C, T, H, W) r%r))rr flipr s rhfliprEis3 ! & &=;<<< 99R==r)r#)F) warningsrwarnrrr"r&r/r8rBrErrrIs  I +++qqq ( $ $ $ 4 4 4&     r