Spaces:
Runtime error
Runtime error
| # for downloading model from google drive | |
| FILE_ID = "17dj0pQieo94TozFv-noSBkXebduij1aM" | |
| MODEL_DIR = 'models' | |
| MODEL_NAME = 'diet_nerf_chair' | |
| class NerfConfig: | |
| # MODEL CONFIG | |
| model = "nerf" | |
| net_activation = "relu" | |
| rgb_activation = "sigmoid" | |
| sigma_activation = "relu" | |
| min_deg_point = 0 | |
| max_deg_point = 10 | |
| deg_view = 4 | |
| # reduce num_coarse_samples, num_fine_samples for speedup | |
| num_coarse_samples = 32 | |
| num_fine_samples = 64 | |
| use_viewdirs = True | |
| near = 2 | |
| far = 6 | |
| noise_std = None | |
| # TODO @Alex: set white_bkgd as flag if we add LLFF dataset | |
| white_bkgd = True | |
| net_depth = 8 | |
| net_width = 256 | |
| net_depth_condition = 1 | |
| net_width_condition = 128 | |
| skip_layer = 4 | |
| num_rgb_channels = 3 | |
| num_sigma_channels = 1 | |
| lindisp = True | |
| legacy_posenc_order = False | |
| randomized = True | |
| # DATA CONFIG | |
| W = 800 | |
| H = 800 | |
| IMAGE_SHAPE = (W, H, 3) | |
| # TODO @Alex: flexible focal if we add LLFF dataset | |
| FOCAL = 555.5555155968841 | |
| # reduce CHUNK if OOM | |
| CHUNK = 4096 | |
| DOWNSAMPLE = 4 | |