Function colorize_alpha

src/sharp/utils/vis.py:39–41

Colorize alpha map.

(alpha: torch.Tensor)

def colorize_alpha(alpha: torch.Tensor) -> torch.Tensor:
    """Colorize alpha map."""
    return colorize_scalar_map(alpha.squeeze(-3), val_min=0.0, val_max=1.0, color_map="coolwarm")


def colorize_scalar_map(