demo/src/config.py

# 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