tink/nerfstudio
4
0 Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-15 20:33:59 +08:00
Code Activity

DyCheck dataset parser (#1270)

Browse Source 
* DyCheck dataset parser

* fix lint
This commit is contained in:
LSong authored and GitHub committed 2023-01-23 13:59:28 -08:00
1 parent 3b8e2ce9ec
commit 9a4f5e8e45
3 files changed
+352 -4
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

nerfstudio/data/utils/data_utils.py
+8 -4
View File
@@ -64,8 +64,8 @@ def get_depth_image_from_path(
interpolation: int = cv2.INTER_NEAREST,
) -> torch.Tensor:
"""Loads, rescales and resizes depth images.
Filepath points to a 16-bit or 32-bit depth image, or a numpy array `*.npy`.
Assumes filepath points to a 16-bit or 32-bit depth image.
Args:
filepath: Path to depth image.
height: Target depth image height.
@@ -76,7 +76,11 @@ def get_depth_image_from_path(
Returns:
Depth image torch tensor with shape [width, height, 1].
"""
image = cv2.imread(str(filepath.absolute()), cv2.IMREAD_ANYDEPTH)
image = image.astype(np.float64) * scale_factor
image = cv2.resize(image, (width, height), interpolation=interpolation)
if filepath.suffix == ".npy":
image = np.load(filepath) * scale_factor
image = cv2.resize(image, (width, height), interpolation=interpolation)
else:
image = cv2.imread(str(filepath.absolute()), cv2.IMREAD_ANYDEPTH)
image = image.astype(np.float64) * scale_factor
image = cv2.resize(image, (width, height), interpolation=interpolation)
return torch.from_numpy(image[:, :, np.newaxis])