| 45 | return group |
| 46 | |
| 47 | class ModelParams(ParamGroup): |
| 48 | def __init__(self, parser, sentinel=False): |
| 49 | self.sh_degree = 3 |
| 50 | self._source_path = "" |
| 51 | self._model_path = "" |
| 52 | self._images = "images" |
| 53 | self._resolution = -1 |
| 54 | self._white_background = False |
| 55 | self.data_device = "cuda" |
| 56 | self.eval = False |
| 57 | self.extension = ".png" |
| 58 | self.num_extra_pts = 0 |
| 59 | self.loaded_pth = "" |
| 60 | self.frame_ratio = 1 |
| 61 | self.dataloader = False |
| 62 | super().__init__(parser, "Loading Parameters", sentinel) |
| 63 | |
| 64 | def extract(self, args): |
| 65 | g = super().extract(args) |
| 66 | g.source_path = os.path.abspath(g.source_path) |
| 67 | return g |
| 68 | |
| 69 | class PipelineParams(ParamGroup): |
| 70 | def __init__(self, parser): |