mirror of
https://github.com/graphdeco-inria/gaussian-splatting
synced 2025-06-26 18:18:11 +00:00
130 lines
4.5 KiB
Python
130 lines
4.5 KiB
Python
#
|
||
# Copyright (C) 2023, Inria
|
||
# GRAPHDECO research group, https://team.inria.fr/graphdeco
|
||
# All rights reserved.
|
||
#
|
||
# This software is free for non-commercial, research and evaluation use
|
||
# under the terms of the LICENSE.md file.
|
||
#
|
||
# For inquiries contact george.drettakis@inria.fr
|
||
#
|
||
|
||
from argparse import ArgumentParser, Namespace
|
||
import sys
|
||
import os
|
||
|
||
class GroupParams:
|
||
pass
|
||
|
||
class ParamGroup:
|
||
def __init__(self, parser: ArgumentParser, name : str, fill_none = False):
|
||
group = parser.add_argument_group(name)
|
||
for key, value in vars(self).items():
|
||
shorthand = False
|
||
if key.startswith("_"):
|
||
shorthand = True
|
||
key = key[1:]
|
||
t = type(value)
|
||
value = value if not fill_none else None
|
||
if shorthand:
|
||
if t == bool:
|
||
group.add_argument("--" + key, ("-" + key[0:1]), default=value, action="store_true")
|
||
else:
|
||
group.add_argument("--" + key, ("-" + key[0:1]), default=value, type=t)
|
||
else:
|
||
if t == bool:
|
||
group.add_argument("--" + key, default=value, action="store_true")
|
||
else:
|
||
group.add_argument("--" + key, default=value, type=t)
|
||
|
||
def extract(self, args):
|
||
group = GroupParams()
|
||
for arg in vars(args).items():
|
||
if arg[0] in vars(self) or ("_" + arg[0]) in vars(self): # 将args中的参数进行分流，分给不同的Params；所以Param初始化中没有的不能额外加入
|
||
setattr(group, arg[0], arg[1]) # 内置函数，将属性 arg[0] 和对应的值 arg[1] 赋给group
|
||
return group
|
||
|
||
class ModelParams(ParamGroup):
|
||
def __init__(self, parser, sentinel=False):
|
||
self.sh_degree = 3
|
||
self._source_path = ""
|
||
self._model_path = ""
|
||
self._images = "images"
|
||
self._depths = ""
|
||
self._resolution = -1
|
||
self._white_background = False
|
||
self.train_test_exp = False
|
||
self.data_device = "cuda"
|
||
self.eval = False
|
||
# xy-new
|
||
self.skybox_locked = False
|
||
self.skybox_num = 0
|
||
self.scaffold_file = ""
|
||
self.bounds_file = ""
|
||
self.is_lidar_depth = False
|
||
self.lidar_depth_max = 30.0
|
||
super().__init__(parser, "Loading Parameters", sentinel) # add parameters into parser
|
||
|
||
def extract(self, args):
|
||
g = super().extract(args)
|
||
g.source_path = os.path.abspath(g.source_path)
|
||
return g
|
||
|
||
class PipelineParams(ParamGroup):
|
||
def __init__(self, parser):
|
||
self.convert_SHs_python = False
|
||
self.compute_cov3D_python = False
|
||
self.debug = False
|
||
self.antialiasing = False
|
||
super().__init__(parser, "Pipeline Parameters")
|
||
|
||
class OptimizationParams(ParamGroup):
|
||
def __init__(self, parser):
|
||
self.iterations = 30_000
|
||
self.position_lr_init = 0.00016
|
||
self.position_lr_final = 0.0000016
|
||
self.position_lr_delay_mult = 0.01
|
||
self.position_lr_max_steps = 30_000
|
||
self.feature_lr = 0.0025
|
||
self.opacity_lr = 0.025
|
||
self.scaling_lr = 0.005
|
||
self.rotation_lr = 0.001
|
||
self.exposure_lr_init = 0.01
|
||
self.exposure_lr_final = 0.001
|
||
self.exposure_lr_delay_steps = 0
|
||
self.exposure_lr_delay_mult = 0.0
|
||
self.percent_dense = 0.01
|
||
self.lambda_dssim = 0.2
|
||
self.densification_interval = 100
|
||
self.opacity_reset_interval = 3000
|
||
self.densify_from_iter = 500
|
||
self.densify_until_iter = 15_000
|
||
self.densify_grad_threshold = 0.0002
|
||
self.depth_l1_weight_init = 1.0
|
||
self.depth_l1_weight_final = 0.01
|
||
self.random_background = False
|
||
self.optimizer_type = "default"
|
||
super().__init__(parser, "Optimization Parameters")
|
||
|
||
def get_combined_args(parser : ArgumentParser):
|
||
cmdlne_string = sys.argv[1:]
|
||
cfgfile_string = "Namespace()"
|
||
args_cmdline = parser.parse_args(cmdlne_string)
|
||
|
||
try:
|
||
cfgfilepath = os.path.join(args_cmdline.model_path, "cfg_args")
|
||
print("Looking for config file in", cfgfilepath)
|
||
with open(cfgfilepath) as cfg_file:
|
||
print("Config file found: {}".format(cfgfilepath))
|
||
cfgfile_string = cfg_file.read()
|
||
except TypeError:
|
||
print("Config file not found at")
|
||
pass
|
||
args_cfgfile = eval(cfgfile_string)
|
||
|
||
merged_dict = vars(args_cfgfile).copy()
|
||
for k,v in vars(args_cmdline).items():
|
||
if v != None:
|
||
merged_dict[k] = v
|
||
return Namespace(**merged_dict)
|