Function array2tensor

detrsmpl/core/renderer/torch3d_renderer/utils.py:53–60

Convert image array to tensor.

(image: np.ndarray)

def array2tensor(image: np.ndarray) -> torch.Tensor:
    """Convert image array to tensor."""
    image = torch.Tensor(image)
    image = normalize(image,
                      origin_value_range=(0, 255),
                      out_value_range=(0, 1),
                      dtype=torch.float32)
    return image


def rgb2bgr(rgbs) -> Union[torch.Tensor, np.ndarray]: