Function tensor2array

detrsmpl/core/renderer/torch3d_renderer/utils.py:43–50

Convert image tensor to array.

(image: torch.Tensor)

def tensor2array(image: torch.Tensor) -> np.ndarray:
    """Convert image tensor to array."""
    image = image.detach().cpu().numpy()
    image = normalize(image,
                      origin_value_range=(0, 1),
                      out_value_range=(0, 255),
                      dtype=np.uint8)
    return image


def array2tensor(image: np.ndarray) -> torch.Tensor: