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

Cameras pr followup (#622)

Browse Source 
cameras manual indexing bug in datamanager
This commit is contained in:
Jake Austin 2022-10-02 22:38:14 -07:00 committed by GitHub
parent 08aaa8b3a5
commit 297950f7ec
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
nerfstudio/cameras/cameras.py
View File

@ -401,7 +401,7 @@ class Cameras:
scaling_factor: Scaling factor to apply to the output resolution.
"""
if isinstance(scaling_factor, float):
scaling_factor = torch.tensor([scaling_factor]).broadcast_to((self.size))
scaling_factor = torch.tensor([scaling_factor]).to(self.device).broadcast_to((self.size))
self.fx = self.fx * scaling_factor
self.fy = self.fy * scaling_factor

12
nerfstudio/data/utils/dataloaders.py
View File

@ -141,13 +141,15 @@ class EvalDataloader(DataLoader):
distortion_params = self.cameras.distortion_params[image_idx]
camera = Cameras(
fx=self.cameras.fx if isinstance(self.cameras.fx, float) else self.cameras.fx[image_idx : image_idx + 1],
fy=self.cameras.fy if isinstance(self.cameras.fy, float) else self.cameras.fy[image_idx : image_idx + 1],
cx=self.cameras.cx,
cy=self.cameras.cy,
fx=self.cameras.fx[image_idx : image_idx + 1],
fy=self.cameras.fy[image_idx : image_idx + 1],
cx=self.cameras.cx[image_idx : image_idx + 1],
cy=self.cameras.cy[image_idx : image_idx + 1],
height=self.cameras.image_height[image_idx : image_idx + 1],
width=self.cameras.image_width[image_idx : image_idx + 1],
camera_to_worlds=self.cameras.camera_to_worlds[image_idx : image_idx + 1],
distortion_params=distortion_params,
camera_type=self.cameras.camera_type,
camera_type=self.cameras.camera_type[image_idx : image_idx + 1],
)
return camera