Function tensor2rgbjet

utils/visualize_utils.py:48–51

(
    tensor: th.Tensor, x_max: Optional[float] = None, x_min: Optional[float] = None
)

    return img

def tensor2rgbjet(
    tensor: th.Tensor, x_max: Optional[float] = None, x_min: Optional[float] = None
) -> np.ndarray:
    return cv2.applyColorMap(tensor2rgb(tensor, x_max=x_max, x_min=x_min), cv2.COLORMAP_JET)


def tensor2rgb(