diff --git a/nerfstudio/data/dataparsers/blender_dataparser.py b/nerfstudio/data/dataparsers/blender_dataparser.py index f1044e826..b365f95cb 100644 --- a/nerfstudio/data/dataparsers/blender_dataparser.py +++ b/nerfstudio/data/dataparsers/blender_dataparser.py @@ -77,7 +77,7 @@ class Blender(DataParser): poses.append(np.array(frame["transform_matrix"])) poses = np.array(poses).astype(np.float32) - img_0 = imageio.imread(image_filenames[0]) + img_0 = imageio.v2.imread(image_filenames[0]) image_height, image_width = img_0.shape[:2] camera_angle_x = float(meta["camera_angle_x"]) focal_length = 0.5 * image_width / np.tan(0.5 * camera_angle_x) diff --git a/nerfstudio/utils/colormaps.py b/nerfstudio/utils/colormaps.py index 19b5922ef..c714f88db 100644 --- a/nerfstudio/utils/colormaps.py +++ b/nerfstudio/utils/colormaps.py @@ -16,8 +16,8 @@ from typing import Literal, Optional +import matplotlib import torch -from matplotlib import cm from torchtyping import TensorType from nerfstudio.utils import colors @@ -88,7 +88,7 @@ def apply_float_colormap( if colormap == "default": colormap = "turbo" - colormap = cm.get_cmap(colormap) + colormap = matplotlib.colormaps[colormap] colormap = torch.tensor(colormap.colors).to(image.device) # type: ignore image = torch.nan_to_num(image, 0) image_long = (image * 255).long() diff --git a/pyproject.toml b/pyproject.toml index 3b4d76647..9b09c6ef9 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -46,7 +46,7 @@ dependencies = [ "requests", "rich>=12.5.1", "scikit-image>=0.19.3", - "tensorboard==2.9.0", + "tensorboard>=2.13.0", "torch>=1.12.1,<2.0.0", "torchmetrics[image]>=0.9.3", "torchtyping>=0.1.4",