MCPcopy Create free account

hub / github.com/VITA-Group/GNT / functions

Functions215 in github.com/VITA-Group/GNT

⨍ Functions 215◇ Types & classes 37
↓ 1 callersFunctionsample_fine_pts
(inv_uniform, N_importance, det, N_samples, ray_batch, weights, z_vals)
gnt/render_ray.py:162
↓ 1 callersMethodsample_random_pixel
(self, N_rand, sample_mode, center_ratio=0.8)
gnt/sample_ray.py:108
↓ 1 callersFunctionsample_target_view_for_training
(views, input_rig_id, input_ids)
gnt/data_loaders/spaces_dataset.py:40
↓ 1 callersMethodsave_model
(self, filename)
gnt/model.py:106
↓ 1 callersFunctionskew
(x)
gnt/data_loaders/data_verifier.py:13
↓ 1 callersFunctionsort_nearby_views_by_angle
(query_pose, ref_poses)
gnt/data_loaders/spaces_dataset.py:90
↓ 1 callersFunctionspherify_poses
(poses, bds)
gnt/data_loaders/llff_data_utils.py:229
↓ 1 callersFunctionspherify_poses
(poses, bds)
gnt/data_loaders/shiny_data_utils.py:205
↓ 1 callersFunctionssim
(img1, img2, window_size=11, size_average=True, format="NCHW")
utils.py:279
↓ 1 callersFunctionssim_utils
(img1, img2, window_size=11, size_average=True)
utils.py:268
↓ 1 callersMethodswitch_to_train
(self)
gnt/model.py:100
↓ 1 callersFunctionsynchronize
Helper function to synchronize (barrier) among all processes when using distributed training
eval.py:25
↓ 1 callersFunctionsynchronize
Helper function to synchronize (barrier) among all processes when using distributed training
render.py:25
↓ 1 callersFunctiontrain
(args)
train.py:43
↓ 1 callersFunctiontwo_view_geometry
:param intrinsics1: 4 by 4 matrix :param extrinsics1: 4 by 4 W2C matrix :param intrinsics2: 4 by 4 matrix :param extrinsics2: 4 by 4
gnt/data_loaders/data_verifier.py:17
↓ 1 callersFunctionvector_norm
Return length, i.e. eucledian norm, of ndarray along axis.
gnt/data_loaders/data_utils.py:14
↓ 1 callersFunctionverify_data
(img1, img2, intrinsics1, extrinsics1, intrinsics2, extrinsics2)
gnt/data_loaders/data_verifier.py:77
↓ 1 callersFunctionview_obj2camera_rgb_path
(view)
gnt/data_loaders/spaces_dataset.py:30
FunctionInterpolateDepths
Returns num_depths from (far_depth, near_depth), interpolated in inv depth. Args: near_depth: The first depth. far_depth: The last
gnt/data_loaders/spaces_dataset.py:195
FunctionReadViewImages
Reads the images for the passed views.
gnt/data_loaders/spaces_dataset.py:218
FunctionWriteNpToImage
Writes an image as a numpy array to the passed path. If the input has more than four channels only the first four will be written. If
gnt/data_loaders/spaces_dataset.py:226
Method__getitem__
(self, idx)
gnt/data_loaders/nerf_synthetic.py:91
Method__getitem__
(self, idx)
gnt/data_loaders/ibrnet_collected.py:73
Method__getitem__
Gets element of the dataset. Args: index: index of the element in the dataset Returns: Single element by index
gnt/data_loaders/create_training_dataset.py:21
Method__getitem__
(self, idx)
gnt/data_loaders/shiny_render.py:88
Method__getitem__
(self, idx)
gnt/data_loaders/llff.py:66
Method__getitem__
(self, idx)
gnt/data_loaders/shiny.py:103
Method__getitem__
(self, idx)
gnt/data_loaders/google_scanned_objects.py:51
Method__getitem__
(self, idx)
gnt/data_loaders/llff_render.py:63
Method__getitem__
(self, idx)
gnt/data_loaders/nerf_synthetic_render.py:92
Method__getitem__
(self, idx)
gnt/data_loaders/spaces_dataset.py:301
Method__getitem__
(self, idx)
gnt/data_loaders/spaces_dataset.py:399
Method__getitem__
(self, idx)
gnt/data_loaders/deepvoxels.py:51
Method__getitem__
(self, indx)
gnt/data_loaders/nmr_dataset.py:107
Method__getitem__
(self, idx)
gnt/data_loaders/realestate.py:68
Method__getitem__
(self, idx)
gnt/data_loaders/llff_test.py:78
Method__init__
(self, window_size=11, size_average=True)
utils.py:243
Method__init__
(self, data, device, resize_factor=1, render_stride=1)
gnt/sample_ray.py:31
Method__init__
(self)
gnt/criterion.py:6
Method__init__
( self, inplanes, planes, stride=1, downsample=None, groups=1,
gnt/feature_network.py:38
Method__init__
( self, inplanes, planes, stride=1, downsample=None, groups=1,
gnt/feature_network.py:93
Method__init__
(self, num_in_layers, num_out_layers, kernel_size, scale)
gnt/feature_network.py:161
Method__init__
( self, encoder="resnet34", coarse_out_ch=32, fine_out_ch=32, norm_lay
gnt/feature_network.py:174
Method__init__
(self, device)
gnt/projection.py:6
Method__init__
(self, **kwargs)
gnt/transformer_network.py:7
Method__init__
(self, dim, hid_dim, dp_rate)
gnt/transformer_network.py:41
Method__init__
(self, dim, dp_rate)
gnt/transformer_network.py:56
Method__init__
(self, dim, ff_hid_dim, ff_dp_rate, attn_dp_rate)
gnt/transformer_network.py:94
Method__init__
(self, dim, n_heads, dp_rate, attn_mode="qk", pos_dim=None)
gnt/transformer_network.py:122
Method__init__
( self, dim, ff_hid_dim, ff_dp_rate, n_heads, attn_dp_rate, attn_mode="qk", pos_dim=None )
gnt/transformer_network.py:176
Method__init__
(self, args, load_opt=True, load_scheduler=True)
gnt/model.py:17
Method__init__
( self, args, mode, # scenes=('chair', 'drum', 'lego', 'hotdog', 'materials',
gnt/data_loaders/nerf_synthetic.py:47
Method__init__
(self, args, mode, random_crop=True, **kwargs)
gnt/data_loaders/ibrnet_collected.py:15
Method__init__
Args: sampler: Sampler used for subsampling num_replicas (int, optional): Number of processes participating in
gnt/data_loaders/create_training_dataset.py:53
Method__init__
(self, args, scenes=(), **kwargs)
gnt/data_loaders/shiny_render.py:14
Method__init__
(self, args, mode, **kwargs)
gnt/data_loaders/llff.py:14
Method__init__
(self, args, mode, scenes=(), random_crop=True, **kwargs)
gnt/data_loaders/shiny.py:14
Method__init__
(self, args, mode, **kwargs)
gnt/data_loaders/google_scanned_objects.py:15
Method__init__
(self, args, scenes="fern", **kwargs)
gnt/data_loaders/llff_render.py:14
Method__init__
( self, args, # scenes=('chair', 'drum', 'lego', 'hotdog', 'materials', 'mic', 'ship')
gnt/data_loaders/nerf_synthetic_render.py:47
Method__init__
Constructor. Args: intrinsics: A numpy 3x3 array representing intrinsics. w_f_c: A numpy 4x4 array representing wFc.
gnt/data_loaders/spaces_dataset.py:109
Method__init__
(self, image_path, shape, camera)
gnt/data_loaders/spaces_dataset.py:130
Method__init__
(self, args, mode, **kwargs)
gnt/data_loaders/spaces_dataset.py:263
Method__init__
(self, args, mode, **kwargs)
gnt/data_loaders/spaces_dataset.py:363
Method__init__
(self, args, subset, scenes="vase", **kwargs)
gnt/data_loaders/deepvoxels.py:14
Method__init__
(self, args, mode, scenes=(), random_crop=True, **kwargs)
gnt/data_loaders/nmr_dataset.py:44
Method__init__
(self, entry)
gnt/data_loaders/realestate.py:11
Method__init__
(self, args, mode, **kwargs)
gnt/data_loaders/realestate.py:41
Method__init__
(self, args, mode, scenes=(), random_crop=True, **kwargs)
gnt/data_loaders/llff_test.py:14
Method__iter__
(self)
gnt/data_loaders/create_training_dataset.py:78
Method__len__
(self)
gnt/data_loaders/nerf_synthetic.py:88
Method__len__
(self)
gnt/data_loaders/ibrnet_collected.py:70
Method__len__
Returns: int: length of the dataset
gnt/data_loaders/create_training_dataset.py:32
Method__len__
(self)
gnt/data_loaders/shiny_render.py:85
Method__len__
(self)
gnt/data_loaders/llff.py:63
Method__len__
(self)
gnt/data_loaders/shiny.py:96
Method__len__
(self)
gnt/data_loaders/google_scanned_objects.py:48
Method__len__
(self)
gnt/data_loaders/llff_render.py:60
Method__len__
(self)
gnt/data_loaders/nerf_synthetic_render.py:89
Method__len__
(self)
gnt/data_loaders/spaces_dataset.py:298
Method__len__
(self)
gnt/data_loaders/spaces_dataset.py:396
Method__len__
(self)
gnt/data_loaders/deepvoxels.py:48
Method__len__
(self)
gnt/data_loaders/nmr_dataset.py:103
Method__len__
(self)
gnt/data_loaders/realestate.py:65
Method__len__
(self)
gnt/data_loaders/llff_test.py:71
Functionbatch_parse_llff_poses
(poses)
gnt/data_loaders/shiny_data_utils.py:28
Functioncalc_angles
(c2w_1, c2w_2)
gnt/data_loaders/data_verifier.py:95
Functionclass_for_name
(module_name, class_name)
gnt/feature_network.py:7
Functioncolor_jitter_all_rgbs
(rgb, ref_rgbs, brightness=0.2, contrast=0.2, saturation=0.2, hue=0.2)
gnt/data_loaders/data_utils.py:155
Functiondilate_img
(img, kernel_size=20)
gnt/sample_ray.py:21
Methodforward
(self, img1, img2)
utils.py:250
Methodforward
training criterion
gnt/criterion.py:9
Methodforward
(self, x)
gnt/feature_network.py:65
Methodforward
(self, x)
gnt/feature_network.py:119
Methodforward
(self, x)
gnt/feature_network.py:156
Methodforward
(self, x)
gnt/feature_network.py:166
Methodforward
(self, x)
gnt/feature_network.py:299
Methodforward
(self, inputs)
gnt/transformer_network.py:36
Methodforward
(self, x)
gnt/transformer_network.py:48
Methodforward
(self, q, k, pos, mask=None)
gnt/transformer_network.py:74
← previousnext →101–200 of 215, ranked by callers