Function resize_to_resolution

flowmap/subsample.py:83–95

(
    image: Float[Tensor, "3 height width"],
    resolution: int,
)

def resize_to_resolution(
    image: Float[Tensor, "3 height width"],
    resolution: int,
) -> Float[Tensor, "3 resized_height resized_width"]:
    _, h, w = image.shape

    scale = (resolution / (h * w)) ** 0.5
    return F.interpolate(
        image[None],
        scale_factor=scale,
        mode="bilinear",
        align_corners=False,
    )[0]


def subsample_frames(