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
↓ 8 callersFunctionchange_default_args
(**kwargs)
former3d/net3d/sparse3d.py:379
↓ 5 callersMethod__init__
(self, input_channels, output_channels, ff_channels, dropout, num_heads, attention_modes, norm)
former3d/net3d/sparse3d.py:100
↓ 4 callersMethodupsample_feats
(self, feats)
former3d/sdfformer.py:331
↓ 3 callersFunctionautocast_norm
(layer_class)
former3d/net3d/sparse3d.py:432
↓ 3 callersFunctionbxyz2bzyx
(indices)
former3d/net3d/sparse3d.py:335
↓ 3 callersFunctionformerTensor2spconvTensor
(former_tensor, spconv_tensor=None)
former3d/net3d/sparse3d.py:10
↓ 3 callersMethodget_volume
(self)
former3d/tsdf_fusion.py:342
↓ 2 callersMethod__init__
(self, in_dim, mlp_dim, out_dim, dropout_rate=0.0)
former3d/transformer.py:10
↓ 2 callersMethod__len__
(self)
former3d/data.py:157
↓ 2 callersMethodback_project_features
(self, bp_data, feats, mv_fuser)
former3d/sdfformer.py:250
↓ 2 callersFunctionbxyz2xyzb
(indices)
former3d/net3d/sparse3d.py:313
↓ 2 callersFunctionbzyx2bxyz
(indices)
former3d/net3d/sparse3d.py:344
↓ 2 callersFunctionconv3x3
3x3 convolution with padding
former3d/net3d/sparse3d.py:503
↓ 2 callersMethoddataloader
(self, split, augment=False)
former3d/lightningmodel.py:137
↓ 2 callersMethodepoch_end
(self, logs, prefix)
former3d/lightningmodel.py:120
↓ 2 callersMethodforward
(self, sp_tensor)
former3d/net3d/sparse3d.py:140
↓ 2 callersMethodget_img_feats
(self, rgb_imgs, proj_mats, cam_positions)
former3d/sdfformer.py:72
↓ 2 callersMethodmesh_opengl
(self, mesh)
scripts/evaluate.py:62
↓ 2 callersFunctionmv_fusion_mean
(features, valid_mask)
former3d/mv_fusion.py:87
↓ 2 callersFunctionnn_correspondance
for each vertex in verts2 find the nearest vertex in verts1 Args: nx3 np.array's Returns: ([indices], [distances])
former3d/eval_utils.py:48
↓ 2 callersMethodproject_voxels
( self, voxel_coords, voxel_batch_inds, projmat, imheight, imwidth )
former3d/sdfformer.py:211
↓ 2 callersFunctionrigid_transform
Applies a rigid transform to an (N, 3) pointcloud.
former3d/tsdf_fusion.py:389
↓ 2 callersFunctionspconvTensor2formerTensor
(spconv_tensor, spvt_tensor)
former3d/net3d/sparse3d.py:24
↓ 2 callersMethodstep
(self, batch, batch_idx)
former3d/lightningmodel.py:60
↓ 2 callersMethodupsample_inds
(self, voxel_inds)
former3d/sdfformer.py:326
↓ 1 callersMethod__getitem__
(self, ind)
former3d/data.py:163
↓ 1 callersMethod__init__
(self, attn_heads, attn_layers, use_proj_occ, voxel_size=0.04)
former3d/sdfformer.py:14
↓ 1 callersMethod__init__
(self, initial_epochs)
former3d/lightningmodel.py:13
↓ 1 callersMethodbuild_map_table
(self)
former3d/net3d/sparse3d.py:71
↓ 1 callersMethodcam2pix
Convert camera coordinates to pixel coordinates.
former3d/tsdf_fusion.py:205
↓ 1 callersFunctioncombineSparseConvTensor
(xs, device)
former3d/net3d/sparse3d.py:348
↓ 1 callersFunctioncompute_proj_occ_loss
(proj_occ_logits, depth_imgs, bp_data, truncation_distance)
former3d/sdfformer.py:354
↓ 1 callersMethodcreate_gather_dict
(self, attention_modes, map_table, voxel_indices, spatial_shape)
former3d/net3d/sparse3d.py:168
↓ 1 callersMethoddelete
(self)
scripts/evaluate.py:65
↓ 1 callersMethoddense
(self, channels_first=True)
former3d/net3d/sparse3d.py:85
↓ 1 callersMethodfix_pose
(self, pose)
scripts/evaluate.py:52
↓ 1 callersFunctionframe_selection
(tiles, pose, intr, imheight, imwidth, n_imgs, rmin_deg, tmin)
scripts/inference.py:104
↓ 1 callersFunctionget_img_feats
(sdfformer, imheight, imwidth, proj_mats, rgb_imgfiles, cam_positions)
scripts/inference.py:136
↓ 1 callersFunctionget_pos_to_kw_map
(func)
former3d/net3d/sparse3d.py:368
↓ 1 callersFunctionget_proj_mats
(intr, pose, factors)
former3d/data.py:60
↓ 1 callersFunctionget_scene_bounds
(pose, intr, imheight, imwidth, frustum_depth)
scripts/inference.py:43
↓ 1 callersFunctionget_subsample_inds
(coords, max_per_example)
former3d/sdfformer.py:340
↓ 1 callersFunctionget_tiles
(minbound, maxbound, cropsize_voxels_fine, voxel_size_fine)
scripts/inference.py:64
↓ 1 callersFunctioninference
(model, info_file, outfile, n_imgs, cropsize, voxel_size=0.04)
scripts/inference.py:167
↓ 1 callersMethodintegrate
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:227
↓ 1 callersMethodintegrate_tsdf
Integrate the TSDF volume.
former3d/tsdf_fusion.py:218
↓ 1 callersFunctionload_model
(ckpt_file, use_proj_occ, config)
scripts/inference.py:18
↓ 1 callersFunctionload_rgb_imgs
(rgb_imgfiles, imheight, imwidth, augment=False)
former3d/data.py:91
↓ 1 callersFunctionload_scene
(info_file)
scripts/inference.py:30
↓ 1 callersFunctionload_tsdf
(tsdf_dir, scene_name)
former3d/data.py:20
↓ 1 callersMethodlosses
(self, voxel_logits, voxel_gt, proj_occ_logits, bp_data, depth_imgs)
former3d/sdfformer.py:163
↓ 1 callersFunctionmake_cuda_ext
(name, module, sources)
setup.py:6
↓ 1 callersFunctionreflect_pose
(pose, plane_pt=None, plane_normal=None)
former3d/data.py:34
↓ 1 callersMethodreset
(self)
former3d/tsdf_fusion.py:597
↓ 1 callersFunctionscatter_nd
pytorch edition of tensorflow scatter_nd. this function don't contain except handle code. so use this carefully when indice repeats, don't sup
former3d/net3d/sparse3d.py:41
↓ 1 callersMethodvox2world
Convert voxel grid coordinates to world coordinates.
former3d/tsdf_fusion.py:193
↓ 1 callersMethodweight_initialization
(self)
former3d/net3d/former_v1.py:186
↓ 1 callersMethodwith_bs_cnt
(self, indices, batch_size)
former3d/net3d/sparse3d.py:126
↓ 1 callersFunctionworker
(arguments)
scripts/evaluate.py:69
↓ 1 callersFunctionwrite_mesh
(outfile, logits_04, voxel_size=0.04)
former3d/lightningmodel.py:163
↓ 1 callersFunctionxyzb2bxyz
(indices)
former3d/net3d/sparse3d.py:304
↓ 1 callersFunctionxyzb2bzyx
(indices)
former3d/net3d/sparse3d.py:322
FunctionPYBIND11_MODULE
former3d/net3d/ops/src/votr_api.cpp:8
Method__call__
(self, height, width, intrinsics, pose, mesh)
scripts/evaluate.py:38
Method__init__
(self, height=480, width=640)
scripts/evaluate.py:33
Method__init__
Constructor. Args: vol_bnds (ndarray): An ndarray of shape (3, 2). Specifies the xyz bounds (min/max) in meters.
former3d/tsdf_fusion.py:10
Method__init__
Constructor. Args: vol_bnds (ndarray): An ndarray of shape (3, 2). Specifies the xyz bounds (min/max) in meters.
former3d/tsdf_fusion.py:548
Method__init__
(self, feat_depth, L)
former3d/view_direction_encoder.py:6
Method__init__
(self, input_depth, n_layers, n_attn_heads, use_var=False)
former3d/mv_fusion.py:12
Method__init__
(self)
former3d/sdfformer.py:303
Method__init__
(self, config)
former3d/lightningmodel.py:44
Method__init__
( self, in_dim, num_heads, mlp_dim, dropout_rate=0.0, attn_dropout_rate=0.0 )
former3d/transformer.py:38
Method__init__
( self, emb_dim, mlp_dim, num_layers=1, num_heads=1, dropout_r
former3d/transformer.py:68
Method__init__
( self, info_files, tsdf_dir, n_imgs, cropsize, voxel_size=0.0
former3d/data.py:136
Method__init__
(self, output_depths, pretrained=True)
former3d/cnn2d.py:6
Method__init__
(self, features, indices, indice_dict, spatial_shape, voxel_size, point_cloud_range, batch_size, hash_size, ma
former3d/net3d/sparse3d.py:58
Method__init__
(self, input_channels, output_channels, ff_channels, dropout, num_heads, attention_modes, norm,
former3d/net3d/sparse3d.py:145
Method__init__
(self, subm_cfg, norm, use_relative_coords=True, use_pooled_feature=True, use_no_query_coords=True, atten_sort
former3d/net3d/sparse3d.py:268
Method__init__
(self, features, indices)
former3d/net3d/sparse3d.py:295
Method__init__
(self, *args, **kw)
former3d/net3d/sparse3d.py:382
Method__init__
(self, input_dim=256, hidden_dim=128, output_dim=32)
former3d/net3d/sparse3d.py:444
Method__init__
(self, inplanes, planes, stride=1, downsam
former3d/net3d/sparse3d.py:468
Method__init__
(self, post_deform=False, **kwargs)
former3d/net3d/former_v1.py:15
Methodbackward
(ctx, a=None)
former3d/net3d/ops/former_utils.py:28
Methodbackward
(ctx, a=None)
former3d/net3d/ops/former_utils.py:52
Methodbackward
(ctx, a=None)
former3d/net3d/ops/former_utils.py:89
Methodbackward
(ctx, a=None)
former3d/net3d/ops/former_utils.py:126
Methodbackward
(ctx, a=None)
former3d/net3d/ops/former_utils.py:155
Methodbackward
(ctx, a=None)
former3d/net3d/ops/former_utils.py:185
Methodbackward
(ctx, a=None)
former3d/net3d/ops/former_utils.py:216
Methodbackward
(ctx, a=None)
former3d/net3d/ops/former_utils.py:248
Methodbackward
(ctx, a=None)
former3d/net3d/ops/former_utils.py:276
Methodbackward
(ctx, a=None)
former3d/net3d/ops/former_utils.py:305
Methodbackward
(ctx, a=None)
former3d/net3d/ops/former_utils.py:334
Methodbackward
(ctx, a=None)
former3d/net3d/ops/former_utils.py:364
Methodbackward
(ctx, a=None)
former3d/net3d/ops/former_utils.py:417
Methodbackward
Args: ctx: grad_out: (M1 + M2 ..., C, nsample) tensor of the gradients of the output from forward Returns:
former3d/net3d/ops/former_utils.py:460
Functionbuild_mapping_with_hash_wrapper
former3d/net3d/ops/src/build_mapping.cpp:57
Functionbuild_mapping_with_tensor_wrapper
former3d/net3d/ops/src/build_mapping.cpp:25
Functionbzyx2xyzb
(indices)
former3d/net3d/sparse3d.py:331
next →1–100 of 183, ranked by callers