mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-13 19:33:46 +08:00
Make nuScenes dataparser compatible with ParallelDataManager (#2635)
The ParallelDataManager (see #2092) makes pytorch crash if the cameras instance's fx, fy, cx or cy tensors are loaded from a common shared tensor. This PR fixes the issue by cloning the respective tensors before passing them to the Cameras(...) constructor.
This commit is contained in:
1 parent
5a772b177d
commit
3d8af87290
1 file changed
+4
-4
@@ -200,10 +200,10 @@ class NuScenes(DataParser):
|
||||
)
|
||||
|
||||
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],
|
||||
|
||||
Reference in new issue
Block a user