From 533abb7d2fbbe0f4c001bba1c1bcafabc8fc9dd2 Mon Sep 17 00:00:00 2001 From: machenmusik Date: Wed, 30 Nov 2022 18:50:48 -0500 Subject: [PATCH] Fix IndexError runtime error (#1057) * Fix IndexError runtime error fixes this error: File "C:\code\nerfstudio\nerfstudio\data\dataparsers\nerfstudio_dataparser.py", line 232, in _generate_dataparser_outputs distortion_params = torch.stack(distort, dim=0)[idx_tensor] IndexError: tensors used as indices must be long, byte or bool tensors * Update nerfstudio_dataparser.py Per discussion at https://github.com/nerfstudio-project/nerfstudio/pull/1057#issuecomment-1332670064 --- nerfstudio/data/dataparsers/nerfstudio_dataparser.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nerfstudio/data/dataparsers/nerfstudio_dataparser.py b/nerfstudio/data/dataparsers/nerfstudio_dataparser.py index 183e7ae8d..e2b86bd6d 100644 --- a/nerfstudio/data/dataparsers/nerfstudio_dataparser.py +++ b/nerfstudio/data/dataparsers/nerfstudio_dataparser.py @@ -212,7 +212,7 @@ class Nerfstudio(DataParser): else: camera_type = CameraType.PERSPECTIVE - idx_tensor = torch.tensor(indices) + idx_tensor = torch.tensor(indices, dtype=torch.long) fx = float(meta["fl_x"]) if fx_fixed else torch.tensor(fx, dtype=torch.float32)[idx_tensor] fy = float(meta["fl_y"]) if fy_fixed else torch.tensor(fy, dtype=torch.float32)[idx_tensor] cx = float(meta["cx"]) if cx_fixed else torch.tensor(cx, dtype=torch.float32)[idx_tensor]