mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-12 02:45:34 +08:00
Fix camera transformations in ARKitScenes and ScanNet (#1544)
* Update arkitscenes_dataparser.py Fix camera coordinate system in ARKitScenes. * Update scannet_dataparser.py Fix camera coordinates in ScanNet.
This commit is contained in:
1 parent
5c810521a5
commit
63f5c8a2c4
2 files changed
+6
-5
No files matched your search
@@ -208,4 +208,6 @@ class ARKitScenes(DataParser):
|
||||
frame_pose = np.array(poses_from_traj[str(my_key)])
|
||||
|
||||
frame_pose[0:3, 1:3] *= -1
|
||||
frame_pose = frame_pose[np.array([1, 0, 2, 3]), :]
|
||||
frame_pose[2, :] *= -1
|
||||
return frame_pose
|
||||
@@ -77,15 +77,14 @@ class ScanNet(DataParser):
|
||||
first_img = cv2.imread(img_dir_sorted[0].as_posix())
|
||||
h, w, _ = first_img.shape
|
||||
|
||||
image_filenames = []
|
||||
depth_filenames = []
|
||||
intrinsics = []
|
||||
poses = []
|
||||
image_filenames, depth_filenames, intrinsics, poses = [], [], [], []
|
||||
|
||||
K = np.loadtxt(self.config.data / "intrinsic" / "intrinsic_color.txt")
|
||||
for img, depth, pose in zip(img_dir_sorted, depth_dir_sorted, pose_dir_sorted):
|
||||
pose = np.loadtxt(pose)
|
||||
pose[0:3, 1:3] *= -1
|
||||
frame_pose[0:3, 1:3] *= -1
|
||||
frame_pose = frame_pose[np.array([1, 0, 2, 3]), :]
|
||||
frame_pose[2, :] *= -1
|
||||
|
||||
# We cannot accept files directly, as some of the poses are invalid
|
||||
if np.isinf(pose).any():
|
||||
|
||||
Reference in new issue
Block a user