e06eee5deec11dc03d6c4975eb9a57f2bc6edd30,kornia/utils/image.py,,image_to_tensor,#Any#,7

Before Change

        torch.Tensor: tensor of the form :math:`(C, H, W)`.

    
    if not type(image) == np.ndarray:
        raise TypeError("Input type is not a numpy.ndarray. Got {}".format(
            type(image)))

    if len(image.shape) > 3 or len(image.shape) < 2:
        raise ValueError("Input size must be a two or three dimensional array")

    tensor: torch.Tensor = torch.from_numpy(image)

    if len(tensor.shape) == 2:
        tensor = torch.unsqueeze(tensor, dim=-1)

    return tensor.permute(2, 0, 1).squeeze_()  // CxHxW

After Change

    input_shape = image.shape
    if len(input_shape) == 2:
        tensor: torch.Tensor = hw_image_to_hw_tensor(image)
    elif len(input_shape) == 3:
        tensor: torch.Tensor = hwc_image_to_chw_tensor(image)
    elif len(input_shape) == 4:
        tensor: torch.Tensor = bhwc_image_to_bchw_tensor(image)
    else:
        raise ValueError("Cannot process image with shape {}".format(input_shape))

Instances