mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-11 02:24:38 +08:00
Add depth_unit_scale_factor to sdfstudio_dataparser (#2285)
Depth supervision for `depth-nerfacto` was added in #1173. This patch extends that support to `sdfstudio` datasets ([as originally intended](https://github.com/nerfstudio-project/nerfstudio/issues/1173#issuecomment-1494633338_)) by adding `depth_unit_scale_factor` to `sdfstudio` metadata.
This commit is contained in:
parent
e6cd95109b
commit
7e8d67f8a8
@ -42,6 +42,8 @@ class SDFStudioDataParserConfig(DataParserConfig):
|
||||
"""Directory specifying location of data."""
|
||||
include_mono_prior: bool = False
|
||||
"""whether or not to load monocular depth and normal """
|
||||
depth_unit_scale_factor: float = 1e-3
|
||||
"""Scales the depth values to meters. Default value is 0.001 for a millimeter to meter conversion."""
|
||||
include_foreground_mask: bool = False
|
||||
"""whether or not to load foreground mask"""
|
||||
downscale_factor: int = 1
|
||||
@ -153,6 +155,7 @@ class SDFStudio(DataParser):
|
||||
# required for normal maps, these are in colmap format so they require c2w before conversion
|
||||
"camera_to_worlds": c2w_colmap if len(c2w_colmap) > 0 else None,
|
||||
"include_mono_prior": self.config.include_mono_prior,
|
||||
"depth_unit_scale_factor": self.config.depth_unit_scale_factor,
|
||||
},
|
||||
)
|
||||
return dataparser_outputs
|
||||
|
||||
Loading…
Reference in New Issue
Block a user