From d9b022b25f53f879a34dd522c5ea5b0fe9cf04f7 Mon Sep 17 00:00:00 2001 From: "J.Y" <132313008+jb-ye@users.noreply.github.com> Date: Fri, 19 Apr 2024 14:03:55 -0400 Subject: [PATCH] Allow alpha transparency training for blender datasets input (#3088) Allow alpha transparency training for rgba input Co-authored-by: Justin Kerr --- nerfstudio/data/dataparsers/blender_dataparser.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/nerfstudio/data/dataparsers/blender_dataparser.py b/nerfstudio/data/dataparsers/blender_dataparser.py index 571799f1d..1027d7504 100644 --- a/nerfstudio/data/dataparsers/blender_dataparser.py +++ b/nerfstudio/data/dataparsers/blender_dataparser.py @@ -41,8 +41,9 @@ class BlenderDataParserConfig(DataParserConfig): """Directory specifying location of data.""" scale_factor: float = 1.0 """How much to scale the camera origins by.""" - alpha_color: str = "white" - """alpha color of background""" + alpha_color: Optional[str] = "white" + """alpha color of background, when set to None, InputDataset that consumes DataparserOutputs will not attempt + to blend with alpha_colors using image's alpha channel data. Thus rgba image will be directly used in training. """ ply_path: Optional[Path] = None """Path to PLY file to load 3D points from, defined relative to the dataset directory. This is helpful for Gaussian splatting and generally unused otherwise. If `None`, points are initialized randomly."""