-
Notifications
You must be signed in to change notification settings - Fork 2
Expand file tree
/
Copy pathopt.py
More file actions
113 lines (104 loc) · 6.27 KB
/
opt.py
File metadata and controls
113 lines (104 loc) · 6.27 KB
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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
import configargparse
def config_parser(cmd=None):
parser = configargparse.ArgumentParser()
parser.add_argument('--config', is_config_file=True,
help='config file path')
parser.add_argument("--expname", type=str, default='scared',
help='experiment name')
parser.add_argument("--basedir", type=str, default='./logs/',
help='where to store ckpts and logs')
parser.add_argument("--datadir", type=str, default='../../Datasets/',
help='input data directory')
parser.add_argument('--with_mvs_guide', action='store_true')
parser.add_argument('--with_rgb_loss', action='store_true')
parser.add_argument('--log', action='store_true')
parser.add_argument('--imgScale_train', type=float, default=0.25)
parser.add_argument('--imgScale_test', type=float, default=0.25)
parser.add_argument('--img_downscale', type=float, default=1.0)
parser.add_argument('--pad', type=int, default=0)
parser.add_argument('--patch_size', type=int, default=6)
parser.add_argument('--patch_num', type=int, default=50)
parser.add_argument('--seed', type=int, default=123456)
# loader options
parser.add_argument("--batch_size", type=int, default=2000)
parser.add_argument("--num_epochs", type=int, default=100)
parser.add_argument("--finetune_epochs", type=int, default=15)
parser.add_argument("--pts_dim", type=int, default=3)
parser.add_argument("--dir_dim", type=int, default=3)
parser.add_argument("--alpha_feat_dim", type=int, default=8)
parser.add_argument('--net_type', type=str, default='v0')
parser.add_argument('--finetune', type=str)
parser.add_argument('--dataset_name', type=str, default='scared',
choices=['hamlyn', 'scared', 'finetune'])
parser.add_argument('--use_color_volume', default=False, action="store_true",
help='project colors into a volume without indexing from image everytime')
parser.add_argument('--use_density_volume', default=False, action="store_true",
help='point sampling with density')
# training options
parser.add_argument("--netdepth", type=int, default=6,
help='layers in network')
parser.add_argument("--view_num", type=int, default=5,
help='layers in network')
parser.add_argument("--netwidth", type=int, default=128,
help='channels per layer')
parser.add_argument("--warmup", type=int, default=128,
help='channels per layer')
parser.add_argument("--netdepth_fine", type=int, default=6,
help='layers in fine network')
parser.add_argument("--netwidth_fine", type=int, default=128,
help='channels per layer in fine network')
parser.add_argument("--lrate", type=float, default=6e-4,
help='learning rate')
parser.add_argument('--decay_step', nargs='+', type=int, default=[8000, 18000, 25000],
help='scheduler decay step')
parser.add_argument('--decay_gamma', type=float, default=0.5,
help='learning rate decay amount')
parser.add_argument('--lr_scheduler', type=str, default='steplr',
help='scheduler type',
choices=['steplr', 'cosine', 'poly'])
parser.add_argument('--warmup_epochs', type=int, default=0,
help='Gradually warm-up(increasing) learning rate in optimizer')
parser.add_argument("--factor", type=int, default=4,
help='number of rays processed in parallel, decrease if running out of memory')
parser.add_argument("--chunk", type=int, default=1024,
help='number of rays processed in parallel, decrease if running out of memory')
parser.add_argument("--netchunk", type=int, default=1024,
help='number of pts sent through network in parallel, decrease if running out of memory')
parser.add_argument("--ckpt", type=str, default=None,
help='specific weights npy file to reload for coarse network')
# rendering options
parser.add_argument('--N_vocab', type=int, default=100,
help='''number of vocabulary (number of images)
in the dataset for nn.Embedding''')
parser.add_argument('--encode_a', default=False, action="store_true",
help='whether to encode appearance (NeRF-A)')
parser.add_argument('--N_a', type=int, default=48,
help='number of embeddings for appearance')
parser.add_argument("--N_samples", type=int, default=90,
help='number of coarse samples per ray')
parser.add_argument('--use_disp', default=False, action="store_true",
help='use disparity depth sampling')
parser.add_argument("--perturb", type=float, default=1.,
help='set to 0. for no jitter, 1. for jitter')
parser.add_argument("--use_viewdirs", default=True,
help='use full 5D input instead of 3D')
parser.add_argument("--i_embed", type=int, default=0,
help='set 0 for default positional encoding, -1 for none')
parser.add_argument("--multires", type=int, default=10,
help='log2 of max freq for positional encoding (3D location)')
parser.add_argument("--multires_views", type=int, default=4,
help='log2 of max freq for positional encoding (2D direction)')
parser.add_argument("--raw_noise_std", type=float, default=0.,
help='std dev of noise added to regularize sigma_a output, 1e0 recommended')
parser.add_argument("--eval", action='store_true',
help='choose to only evaluate the model')
## blender flags
parser.add_argument("--white_bkgd", action='store_true',
help='set to render synthetic data on a white bkgd (always use for dvoxels)')
# logging/saving options
parser.add_argument("--N_vis", type=int, default=20,
help='frequency of visualize the depth')
if cmd is not None:
return parser.parse_args(cmd)
else:
return parser.parse_args()