DietNerf-Demo / demo /src /config.py
alexlau's picture
set DOWNSAMPLE=4 and then resize a bigger image
1a30119
raw history blame
No virus
1.08 kB
# 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