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

Fixing resize_image and adding additional distortion check (#3162)

Browse Source 
adding fixes for the runtime error on masking as well as distortion checking

Co-authored-by: J.Y <132313008+jb-ye@users.noreply.github.com>
This commit is contained in:
Anthony Zhang 2024-05-24 20:31:45 -07:00 committed by GitHub
parent 5a341eac2c
commit 456930c213
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
nerfstudio/data/datamanagers/full_images_datamanager.py
View File

@ -161,7 +161,7 @@ class FullImageDatamanager(DataManager, Generic[TDataset]):
f'The size of image ({data["image"].shape[1]}, {data["image"].shape[0]}) loaded '
f'does not match the camera parameters ({camera.width.item(), camera.height.item()})'
)
if camera.distortion_params is None:
if camera.distortion_params is None or torch.all(camera.distortion_params == 0):
return data
K = camera.get_intrinsics_matrices().numpy()
distortion_params = camera.distortion_params.numpy()

1
nerfstudio/models/splatfacto.py
View File

@ -91,6 +91,7 @@ def resize_image(image: torch.Tensor, d: int):
"""
import torch.nn.functional as tf
image = image.to(torch.float32)
weight = (1.0 / (d * d)) * torch.ones((1, 1, d, d), dtype=torch.float32, device=image.device)
return tf.conv2d(image.permute(2, 0, 1)[:, None, ...], weight, stride=d).squeeze(1).permute(1, 2, 0)