Function resize_ims

diff2flow/helpers.py:158–160

(x: Tensor, size: int, mode: str = "bilinear", **kwargs)

def resize_ims(x: Tensor, size: int, mode: str = "bilinear", **kwargs):
    # idea: blur image before down-sampling
    return nn.functional.interpolate(x, size=size, mode=mode, **kwargs)


def get_batch_stats(x: Tensor, stats=None):