khlddlZddlZejddZdZdZd dZdZdZd d Z d Z y) 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|j dk(st d|j zy)Nzclip should be Tensor. Got %szclip should be 4D. Got %dDT)torch is_tensor TypeErrortype ndimension ValueErrordimclips d/var/www/teggl/fontify/venv/lib/python3.12/site-packages/torchvision/transforms/_functional_video.py_is_tensor_video_clipr sL ??4 7$t*DEE ??  !5 BCC ctt|jdk7r td|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 rcroprsE  499;1566 QQYAE ) **rct|dk7rtd|tjjj |||dS)Nz9target size should be tuple (height, width), instead got F)rmode align_corners)rr rnn functional interpolate)r target_sizeinterpolation_modes rresizer" sJ ;1TU`Tabcc 88   * *4kHZjo * pprcjt|s tdt|||||}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%&s> ! &;<< aAq !D $0 1D Krc,t|s td|jd|jd}}|\}}||ks||kr tdtt ||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 FB2vRMNN E1r6S. !"A E1r6S. !"A aB ##rct||jtjk(s!t dt |jz|j jdddddz 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_tensorr7GsZ$ :: $ICPTPZPZO[\\ ::<  1a +e 33rcjt|s td|s|j}tj||j |j }tj||j |j }|j|dddddfj|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$)r2deviceN) rr cloner as_tensorr2r9sub_div_)r meanstdinplaces r normalizerAVs ! &;<< zz| ??4tzz$++ FD //#TZZ DCIId1dD$&'(--c!T42E.FG KrcPt|s td|jdS)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 rhfliprDis& ! &;<< 99R=r)bilinear)F) warningsrwarnrrr"r%r.r7rArDrrrIsJ  I +q ( $ 4& r