From 6270e6b085417ac895c4d4ebe227a2038f0bd672 Mon Sep 17 00:00:00 2001 From: Justin Kerr Date: Mon, 13 Jan 2025 12:54:52 -0800 Subject: [PATCH] remove profiling --- nerfstudio/data/utils/data_utils.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/nerfstudio/data/utils/data_utils.py b/nerfstudio/data/utils/data_utils.py index 97f7a53f0f..e79f169034 100644 --- a/nerfstudio/data/utils/data_utils.py +++ b/nerfstudio/data/utils/data_utils.py @@ -22,7 +22,6 @@ import torch from PIL import Image from PIL.Image import Image as PILImage -from torch.profiler import record_function def pil_to_numpy(im: PILImage) -> np.ndarray: @@ -35,8 +34,7 @@ def pil_to_numpy(im: PILImage) -> np.ndarray: numpy.ndarray representing the image data. """ # Load in image completely (PIL defaults to lazy loading) - with record_function("im.load()"): - im.load() + im.load() # Unpack data e = Image._getencoder(im.mode, "raw", im.mode)