File size: 1,080 Bytes
19677a1
7d3d587
0d35ba8
7d3d587
19677a1
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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