Spaces:
Runtime error
Runtime error
File size: 1,080 Bytes
19677a1 7d3d587 0d35ba8 7d3d587 19677a1 12def95 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 |
# 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 = 2
|