MCPcopy Create free account

hub / github.com/apchenstu/mvsnerf / functions

Functions176 in github.com/apchenstu/mvsnerf

⨍ Functions 176◇ Types & classes 25
Method__init__
spheric_poses: whether the images are taken in a spheric inward-facing manner default: False (forward-facing)
data/llff.py:158
Method__init__
img_wh should be set to a tuple ex: (1152, 864) to enable test mode!
data/dtu.py:23
Method__init__
img_wh should be set to a tuple ex: (1152, 864) to enable test mode!
data/dtu_ft.py:12
Method__init__
(self, args, split='train', load_ref=False)
data/blender.py:13
Method__len__
(self)
data/llff.py:362
Method__len__
(self)
data/dtu.py:134
Method__len__
(self)
data/dtu_ft.py:195
Method__len__
(self)
data/blender.py:160
Method_upsample_add
(self, x, y)
models.py:711
Functionabs_error_numpy
(depth_pred, depth_gt, mask)
utils.py:67
Methodbuild_volume_costvar
(self, feats, proj_mats, depth_values, pad=0)
models.py:787
Functioncolorjitter
(img, factor)
data/dtu.py:12
Methodconfigure_optimizers
(self)
train_mvs_nerf_fusion_finetuning_pl.py:232
Methodconfigure_optimizers
(self)
train_mvs_nerf_pl.py:84
Methodconfigure_optimizers
(self)
train_mvs_nerf_finetuning_pl.py:117
Functioncreate_spheric_poses
Create circular poses around z axis. Inputs: radius: the (negative) height and the radius of the circle. Outputs: spheric
data/llff.py:116
Functioncreate_spiral_poses
Computes poses that follow a spiral path for rendering purpose. See https://github.com/Fyusion/LLFF/issues/19 In particular, the path loo
data/llff.py:83
Functiondetach_data
(dictionary)
utils.py:434
Methodforward
(self, q, k, v, mask=None)
models.py:78
Methodforward
(self, q, k, v, mask=None)
models.py:112
Methodforward
(self, x)
models.py:194
Methodforward
(self, x)
models.py:273
Methodforward
(self, x)
models.py:341
Methodforward
(self, x)
models.py:421
Methodforward
(self, x)
models.py:510
Methodforward
(self, x)
models.py:565
Methodforward
(self, x)
models.py:671
Methodforward
(self, x)
models.py:684
Methodforward
(self, x)
models.py:715
Methodforward
(self, x)
models.py:756
Methodforward
(self, imgs, proj_mats, near_far, pad=0, return_color=False, lindisp=False)
models.py:895
Methodforward
coordinate: [N, 3] z,x,y
models.py:941
Methodforward
(self, depth_pred, depth_gt, mask=None)
train_mvs_nerf_fusion_finetuning_pl.py:28
Methodforward
(self)
train_mvs_nerf_fusion_finetuning_pl.py:228
Methodforward
(self, depth_pred, depth_gt, mask=None)
train_mvs_nerf_pl.py:28
Methodforward
(self)
train_mvs_nerf_pl.py:74
Methodforward
(self, depth_pred, depth_gt, mask=None)
train_mvs_nerf_finetuning_pl.py:26
Methodforward
(self)
train_mvs_nerf_finetuning_pl.py:113
Methodforward_alpha
(self, x)
models.py:176
Methodforward_alpha
(self,x)
models.py:258
Methodforward_alpha
(self,x)
models.py:495
Functiongen_angle_feature
Inputs: c2ws: [1,v,4,4] rays_pts: [N_rays, N_samples, 3] rays_dir: [N_rays, 3] Returns:
renderer.py:96
Functiongen_render_path
(c2ws, N_views=30)
utils.py:479
Functiongen_render_path_pixelNeRF
(c2ws, N_views=30)
utils.py:541
Functiongen_render_path_spherical
(theta, phi, radius=1.0)
utils.py:528
Methodget_lr
(self)
train_mvs_nerf_finetuning_pl.py:122
Functionget_nearest_pose_ids
Args: tar_pose: target pose [N, 4, 4] ref_poses: reference poses [M, 4, 4] num_select: the number of nearest views to sel
utils.py:698
Functionget_psnr
(imgs_pred, imgs_gt)
utils.py:17
Methodload_poses_all
(self, file=f"transforms_train.json")
data/blender.py:148
Functionndc2dist
(ndc_pts, cos_angle)
renderer.py:13
Functionpose_spherical_dtu
Computes poses that follow a spiral path for rendering purpose. See https://github.com/Fyusion/LLFF/issues/19 In particular, the path loo
utils.py:644
Functionpose_spherical_nerf
(euler, radius=4.0)
utils.py:634
Methodprepare_data
(self)
train_mvs_nerf_pl.py:77
Methodread_source_views
(self, pair_idx=None, device=torch.device("cpu"))
data/llff.py:274
Methodread_source_views
(self, file=f"transforms_train.json", pair_idx=None, device=torch.device("cpu"))
data/blender.py:91
Functionret
(inputs, alpha_only)
renderer.py:34
Functiontensor_unsqueeze
(data, filter)
utils.py:410
Functionto_cuda
(data, device, filter)
utils.py:402
Functionto_tensor_cuda
(data, device, filter)
utils.py:389
Methodtrain_dataloader
(self)
train_mvs_nerf_fusion_finetuning_pl.py:237
Methodtrain_dataloader
(self)
train_mvs_nerf_pl.py:90
Methodtrain_dataloader
(self)
train_mvs_nerf_finetuning_pl.py:126
Methodtraining_step
(self, batch, batch_nb)
train_mvs_nerf_fusion_finetuning_pl.py:251
Methodtraining_step
(self, batch, batch_nb)
train_mvs_nerf_pl.py:104
Methodtraining_step
(self, batch, batch_nb)
train_mvs_nerf_finetuning_pl.py:140
Methodval_dataloader
(self)
train_mvs_nerf_fusion_finetuning_pl.py:244
Methodval_dataloader
(self)
train_mvs_nerf_pl.py:97
Methodval_dataloader
(self)
train_mvs_nerf_finetuning_pl.py:133
Methodvalidation_epoch_end
(self, outputs)
train_mvs_nerf_fusion_finetuning_pl.py:345
Methodvalidation_epoch_end
(self, outputs)
train_mvs_nerf_pl.py:256
Methodvalidation_epoch_end
(self, outputs)
train_mvs_nerf_finetuning_pl.py:254
Methodvalidation_step
(self, batch, batch_nb)
train_mvs_nerf_fusion_finetuning_pl.py:294
Methodvalidation_step
(self, batch, batch_nb)
train_mvs_nerf_pl.py:172
Methodvalidation_step
(self, batch, batch_nb)
train_mvs_nerf_finetuning_pl.py:192
Functionvisualize_depth_numpy
depth: (H, W)
utils.py:30
Functionweights_init
(m)
models.py:10
← previous101–176 of 176, ranked by callers