tink/nerfstudio
4
0
Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-11 10:34:29 +08:00
Code

Fix pose centering flag for ARKitScenes dataparser (#1528)

Browse Source
This commit is contained in:
Georgy Ponimatkin 2023-02-28 02:30:53 +01:00 committed by GitHub
parent 258ee26983
commit 2eda6b4523
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
nerfstudio/data/dataparsers/arkitscenes_dataparser.py
View File

@ -141,7 +141,7 @@ class ARKitScenes(DataParser):
poses = torch.from_numpy(np.stack(poses).astype(np.float32))
intrinsics = torch.from_numpy(np.stack(intrinsics).astype(np.float32))
if self.config.scale_poses:
if self.config.center_poses:
poses[:, :3, 3] -= poses[:, :3, 3].mean(dim=0)
if self.config.scale_poses: