Skip to content

Commit

Permalink
Make nuScenes dataparser compatible with ParallelDataManager
Browse files Browse the repository at this point in the history
The ParallelDataManager (see nerfstudio-project#2092) makes pytorch crash if the cameras instance's fx, fy, cx or cy
tensors are loaded from a common shared tensor
  • Loading branch information
dmholtz committed Nov 29, 2023
1 parent c896ee4 commit a405377
Showing 1 changed file with 5 additions and 9 deletions.
14 changes: 5 additions & 9 deletions nerfstudio/data/dataparsers/nuscenes_dataparser.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,11 +25,7 @@
from nuscenes.nuscenes import NuScenes as NuScenesDatabase

from nerfstudio.cameras.cameras import Cameras, CameraType
from nerfstudio.data.dataparsers.base_dataparser import (
DataParser,
DataParserConfig,
DataparserOutputs,
)
from nerfstudio.data.dataparsers.base_dataparser import DataParser, DataParserConfig, DataparserOutputs
from nerfstudio.data.scene_box import SceneBox


Expand Down Expand Up @@ -200,10 +196,10 @@ def _generate_dataparser_outputs(self, split="train"):
)

cameras = Cameras(
fx=intrinsics[:, 0, 0],
fy=intrinsics[:, 1, 1],
cx=intrinsics[:, 0, 2],
cy=intrinsics[:, 1, 2],
fx=intrinsics[:, 0, 0].detach().clone(),
fy=intrinsics[:, 1, 1].detach().clone(),
cx=intrinsics[:, 0, 2].detach().clone(),
cy=intrinsics[:, 1, 2].detach().clone(),
height=900,
width=1600,
camera_to_worlds=poses[:, :3, :4],
Expand Down

0 comments on commit a405377

Please sign in to comment.