Function crop

datasets/video_utils.py:48–55

Args: clip (torch.tensor): Video clip to be cropped. Size is (T, C, H, W)

(clip, i, j, h, w)

def crop(clip, i, j, h, w):
    """
    Args:
        clip (torch.tensor): Video clip to be cropped. Size is (T, C, H, W)
    """
    if len(clip.size()) != 4:
        raise ValueError("clip should be a 4D tensor")
    return clip[..., i : i + h, j : j + w]


def resize(clip, target_size, interpolation_mode):