e06eee5deec11dc03d6c4975eb9a57f2bc6edd30,kornia/utils/image.py,,tensor_to_image,#Any#,32

Before Change

    if len(input_shape) == 2:
        tensor = torch.squeeze(tensor, dim=-1)

    return tensor.cpu().detach().numpy()

After Change

    input_shape = tensor.shape
    if len(input_shape) == 2:
        image: np.array = hw_tensor_to_hw_image(tensor)
    elif len(input_shape) == 3:
        image: np.array = chw_tensor_to_hwc_image(tensor)
    elif len(input_shape) == 4:
        image: np.array = bchw_tensor_to_bhwc_image(tensor)
    else:
        raise ValueError("Cannot process tensor with shape {}".format(input_shape))

    return image

Instances