Function any2tensor

core/utils/any2tensor.py:36–50

Convert strpath, PIL.Image.Image, numpy.ndarray, torch.Tensor image(s) to a torch.Tensor. Args: imgs (list[strpath] | list[PIL.Image.Image] | list[numpy.ndarray] | list[torch.Tensor] | strpath | PIL.Image.Image | numpy.ndarray | torch.Tensor): The input images. Returns: tor

(imgs)

def any2tensor(imgs):
    """Convert strpath, PIL.Image.Image, numpy.ndarray, torch.Tensor image(s) to a torch.Tensor.

    Args:
        imgs (list[strpath] | list[PIL.Image.Image] | list[numpy.ndarray] | list[torch.Tensor] | strpath | PIL.Image.Image | numpy.ndarray | torch.Tensor): The input images.

    Returns:
        torch.Tensor: The converted image(s).
    """
    if isinstance(imgs, list):
        return torch.stack([_any2tensor(img) for img in imgs], dim=0)
    elif isinstance(imgs, (str, PIL.Image.Image, numpy.ndarray, torch.Tensor)):
        return _any2tensor(imgs)
    else:
        raise TypeError('imgs is an unsupported type, imgs should be list[strpath] | list[PIL.Image.Image] | list[numpy.ndarray] | list[torch.Tensor] | strpath | PIL.Image.Image | numpy.ndarray | torch.Tensor. But got {}'.format(type(imgs)))