diff --git a/nerfstudio/utils/colormaps.py b/nerfstudio/utils/colormaps.py index 0e09c0bfd..d772d40b4 100644 --- a/nerfstudio/utils/colormaps.py +++ b/nerfstudio/utils/colormaps.py @@ -24,7 +24,7 @@ from torch import Tensor from nerfstudio.utils import colors -Colormaps = Literal["default", "turbo", "viridis", "magma", "inferno", "cividis", "pca"] +Colormaps = Literal["default", "turbo", "viridis", "magma", "inferno", "cividis", "gray", "pca"] @dataclass(frozen=True) @@ -104,6 +104,8 @@ def apply_float_colormap(image: Float[Tensor, "*bs 1"], colormap: Colormaps = "v colormap = "turbo" image = torch.nan_to_num(image, 0) + if colormap == "gray": + return image.repeat(1, 1, 3) image_long = (image * 255).long() image_long_min = torch.min(image_long) image_long_max = torch.max(image_long) diff --git a/nerfstudio/viewer/server/control_panel.py b/nerfstudio/viewer/server/control_panel.py index 456e3598c..7039b61c3 100644 --- a/nerfstudio/viewer/server/control_panel.py +++ b/nerfstudio/viewer/server/control_panel.py @@ -361,7 +361,7 @@ def _get_colormap_options(dimensions: int, dtype: type) -> List[Colormaps]: if dimensions == 3: colormap_options = ["default"] if dimensions == 1 and dtype == torch.float: - colormap_options = [c for c in list(get_args(Colormaps)) if c != "default"] + colormap_options = [c for c in list(get_args(Colormaps)) if c not in ("default", "pca")] if dimensions > 3: colormap_options = ["pca"] return colormap_options