MCPcopy Create free account

hub / github.com/alibaba-damo-academy/former3d / functions

Functions183 in github.com/alibaba-damo-academy/former3d

⨍ Functions 183◇ Types & classes 40
Methodconfigure_optimizers
(self)
former3d/lightningmodel.py:51
Functionconv1x1
1x1 convolution
former3d/net3d/sparse3d.py:515
Functiondownsample_with_hash_wrapper
former3d/net3d/ops/src/build_mapping.cpp:71
Functiondownsample_with_tensor_wrapper
former3d/net3d/ops/src/build_mapping.cpp:39
Functioneval_depth
Computes 2d metrics between two depth maps Args: depth_pred: mxn np.array containing prediction depth_trgt: mxn np.array containi
former3d/eval_utils.py:116
Functioneval_mesh
Compute Mesh metrics between prediction and target. Opens the Meshs and runs the metrics Args: file_pred: file path of prediction
former3d/eval_utils.py:5
Methodfinetune_function
(self, pl_module, current_epoch, optimizer, optimizer_idx)
former3d/lightningmodel.py:26
Methodforward
(self, feats, proj, cam_positions)
former3d/view_direction_encoder.py:17
Methodforward
(self, features, valid_mask)
former3d/mv_fusion.py:7
Methodforward
(self, features, bp_depth, bp_mask, use_proj_occ, var_imgs=None)
former3d/mv_fusion.py:31
Methodforward
(self, batch, voxel_inds_16)
former3d/sdfformer.py:82
Methodforward
(self, x)
former3d/transformer.py:25
Methodforward
(self, x, mask=None)
former3d/transformer.py:52
Methodforward
(self, x, mask=None)
former3d/transformer.py:89
Methodforward
(self, x)
former3d/cnn2d.py:58
Methodforward
(self, sp_tensor)
former3d/net3d/sparse3d.py:191
Methodforward
(self, sp_tensor)
former3d/net3d/sparse3d.py:286
Methodforward
(self, input)
former3d/net3d/sparse3d.py:434
Methodforward
(self, x)
former3d/net3d/sparse3d.py:457
Methodforward
(self, x)
former3d/net3d/sparse3d.py:483
Methodforward
(self, input, voxel_dim, res, hash_size)
former3d/net3d/former_v1.py:192
Methodforward
Args: ctx: voxel_indices: (num_voxels, 4) (bs_idx, z, y, x) Returns:
former3d/net3d/ops/former_utils.py:11
Methodforward
Args: ctx: voxel_indices: (num_voxels, 4) (bs_idx, z, y, x) Returns:
former3d/net3d/ops/former_utils.py:36
Methodforward
Args: ctx: voxel_indices: (num_voxels, 4) (bs_idx, z, y, x) Returns:
former3d/net3d/ops/former_utils.py:59
Methodforward
Args: ctx: voxel_indices: (num_voxels, 4) (bs_idx, z, y, x) Returns:
former3d/net3d/ops/former_utils.py:96
Methodforward
Args: ctx: dense_map: (bs_idx, x_max, y_max, z_max) -> old map table voxel_indices: (num_voxels, 4) (bs_i
former3d/net3d/ops/former_utils.py:134
Methodforward
Args: ctx: dense_map: (bs_idx, hash_size, 2) -> old map table voxel_indices: (num_voxels, 4) (bs_idx, z,
former3d/net3d/ops/former_utils.py:163
Methodforward
Args: ctx: dense_map: (bs_idx, x_max, y_max, z_max) -> old map table voxel_indices: (num_voxels, 4) (bs_i
former3d/net3d/ops/former_utils.py:193
Methodforward
Args: ctx: dense_map: (bs_idx, x_max, y_max, z_max) -> old map table voxel_indices: (num_voxels, 4) (bs_i
former3d/net3d/ops/former_utils.py:224
Methodforward
Args: ctx: voxel_indices: (num_voxels, 4) (bs_idx, z, y, x) Returns:
former3d/net3d/ops/former_utils.py:256
Methodforward
Args: ctx: voxel_indices: (num_voxels, 4) (bs_idx, z, y, x) Returns:
former3d/net3d/ops/former_utils.py:284
Methodforward
Args: ctx: voxel_indices: (num_voxels, 4) (bs_idx, z, y, x) Returns:
former3d/net3d/ops/former_utils.py:313
Methodforward
Args: ctx: voxel_indices: (num_voxels, 4) (bs_idx, z, y, x) Returns:
former3d/net3d/ops/former_utils.py:342
Methodforward
Args: ctx: voxel_indices: (num_voxels, 4) (bs_idx, z, y, x) Returns:
former3d/net3d/ops/former_utils.py:373
Methodforward
Args: ctx: features: (N1 + N2 ..., C) tensor of features to group features_batch_cnt: (batch_size) [N1 +
former3d/net3d/ops/former_utils.py:426
Functionframe_selection
( poses, intr, imwidth, imheight, sample_pts, tmin, rmin_deg, n_imgs, )
former3d/utils.py:62
Methodfreeze_before_training
(self, pl_module)
former3d/lightningmodel.py:17
Methodget_mesh
Compute a mesh from the voxel volume using marching cubes.
former3d/tsdf_fusion.py:369
Methodget_point_cloud
Extract a point cloud from the voxel volume.
former3d/tsdf_fusion.py:349
Functionget_view_frustum
Get corners of 3D camera view frustum of depth image
former3d/tsdf_fusion.py:396
Methodget_volume
(self)
former3d/tsdf_fusion.py:631
Methodgetitem
(self, ind, **kwargs)
former3d/data.py:160
Functionglobal_avg_pool
(inputs: spconv.SparseConvTensor)
former3d/net3d/sparse3d.py:421
Functiongroup_features_grad_wrapper_stack
former3d/net3d/ops/src/group_features.cpp:25
Functiongroup_features_wrapper_stack
former3d/net3d/ops/src/group_features.cpp:46
Functionintegrate
( depth_im, cam_intr, cam_pose, obs_weight, world_c, vox_coords, weight_vol, t
former3d/tsdf_fusion.py:493
Methodintegrate
Integrate an RGB-D frame into the TSDF volume. Args: color_im (ndarray): An RGB image of shape (H, W, 3). depth_im (ndarra
former3d/tsdf_fusion.py:602
Functionlayer_wrapper
(layer_class)
former3d/net3d/sparse3d.py:380
Functionload_info_files
(scannet_dir, split)
former3d/utils.py:115
Functionlog_transform
rescales TSDF values to weight voxels near the surface more than close to the truncation distance
former3d/utils.py:13
Functionmeshwrite
Save a 3D mesh to a polygon .ply file.
former3d/tsdf_fusion.py:416
Methodon_train_epoch_start
(self)
former3d/lightningmodel.py:57
Methodon_train_start
(self, trainer, pl_module)
scripts/train.py:15
Methodon_validation_end
(self, trainer, pl_module)
scripts/train.py:19
Methodon_validation_epoch_start
(self)
former3d/lightningmodel.py:91
Methodon_validation_start
(self, trainer, pl_module)
scripts/train.py:17
Functionpcwrite
Save a point cloud to a polygon .ply file.
former3d/tsdf_fusion.py:460
Functionproject_to_mesh
Transfers attributs from from_mesh to to_mesh using nearest neighbors Each vertex in to_mesh gets assigned the attribute of the nearest vertex
former3d/eval_utils.py:75
Functionremove_redundant
(poses, rmin_deg, tmin)
former3d/utils.py:48
Methodsdf_trunc
(self)
former3d/tsdf_fusion.py:635
Functionsparse_collate_fn
(batch)
former3d/collate.py:5
Functionsparse_local_attention_with_hash_wrapper
former3d/net3d/ops/src/build_attention_indices.cpp:41
Functionsparse_local_attention_with_tensor_wrapper
former3d/net3d/ops/src/build_attention_indices.cpp:25
Functionsparse_strided_attention_with_hash_wrapper
former3d/net3d/ops/src/build_attention_indices.cpp:106
Functionsparse_strided_attention_with_tensor_wrapper
former3d/net3d/ops/src/build_attention_indices.cpp:87
Functionsparsify
(volume, valid, viz=False)
former3d/net3d/sparse3d.py:395
Functionsubm_local_attention_with_hash_wrapper
former3d/net3d/ops/src/build_attention_indices.cpp:72
Functionsubm_local_attention_with_tensor_wrapper
former3d/net3d/ops/src/build_attention_indices.cpp:57
Functionsubm_strided_attention_with_hash_sort_wrapper
former3d/net3d/ops/src/build_attention_indices.cpp:161
Functionsubm_strided_attention_with_hash_wrapper
former3d/net3d/ops/src/build_attention_indices.cpp:143
Functionsubm_strided_attention_with_tensor_wrapper
former3d/net3d/ops/src/build_attention_indices.cpp:125
Functionto_mesh
(vol, voxel_size=1, origin=np.zeros(3), level=0, mask=None)
former3d/utils.py:27
Functionto_vol
(inds, vals)
former3d/utils.py:20
Functiontoggle_geom
(vis, geom_ind)
former3d/utils.py:129
Methodtrain_dataloader
(self)
former3d/lightningmodel.py:131
Methodtraining_epoch_end
(self, outputs)
former3d/lightningmodel.py:114
Methodtraining_step
(self, batch, batch_idx)
former3d/lightningmodel.py:79
Methodval_dataloader
(self)
former3d/lightningmodel.py:134
Methodvalidation_epoch_end
(self, outputs)
former3d/lightningmodel.py:117
Methodvalidation_step
(self, batch, batch_idx)
former3d/lightningmodel.py:94
Functionvisualize
(o3d_geoms)
former3d/utils.py:126
Methodvoxel_size
(self)
former3d/tsdf_fusion.py:639
Methodwith_coords
(self, indices, point_cloud_range, voxel_size)
former3d/net3d/sparse3d.py:134
← previous101–183 of 183, ranked by callers