tink/nerfstudio
4
0 Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-13 03:14:00 +08:00
Code Activity

fix bugs in generfacto, train_dataparser_outputs.save_dataparser_transform and viewer camera on cpu (#3378)

Browse Source 
* fix bugs in generfacto, train_dataparser_outputs.save_dataparser_transform and viewer camera on cpu

* fix formating
This commit is contained in:
pierremerriaux-leddartech authored and GitHub committed 2024-08-23 11:44:47 -04:00
1 parent 36d5e35801
commit e3a506f346
2 files changed
+6 -6
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

nerfstudio/engine/trainer.py
+4 -4
View File
@@ -229,10 +229,10 @@ class Trainer:
def train(self) -> None:
"""Train the model."""
assert self.pipeline.datamanager.train_dataset is not None, "Missing DatsetInputs"
self.pipeline.datamanager.train_dataparser_outputs.save_dataparser_transform(
self.base_dir / "dataparser_transforms.json"
)
if hasattr(self.pipeline.datamanager, "train_dataparser_outputs"):
self.pipeline.datamanager.train_dataparser_outputs.save_dataparser_transform(
self.base_dir / "dataparser_transforms.json"
)
self._init_viewer_state()
with TimeWriter(writer, EventName.TOTAL_TRAIN_TIME):
nerfstudio/viewer/viewer.py
+2 -2
View File
@@ -458,9 +458,9 @@ class Viewer:
R = R @ vtf.SO3.from_x_radians(np.pi)
camera_handle = self.viser_server.scene.add_camera_frustum(
name=f"/cameras/camera_{idx:05d}",
fov=float(2 * np.arctan(camera.cx / camera.fx[0])),
fov=float(2 * np.arctan((camera.cx / camera.fx[0]).cpu())),
scale=self.config.camera_frustum_scale,
aspect=float(camera.cx[0] / camera.cy[0]),
aspect=float((camera.cx[0] / camera.cy[0]).cpu()),
image=image_uint8,
wxyz=R.wxyz,
position=c2w[:3, 3] * VISER_NERFSTUDIO_SCALE_RATIO,