MCPcopy Create free account

hub / github.com/facebookresearch/sam-3d-objects / functions

Functions784 in github.com/facebookresearch/sam-3d-objects

⨍ Functions 784◇ Types & classes 121
Methodforward
(self, input: SparseTensor)
sam3d_objects/model/backbone/tdfy_dit/modules/sparse/norm.py:38
Methodforward
(self, x: SparseTensor)
sam3d_objects/model/backbone/tdfy_dit/modules/sparse/norm.py:54
Methodforward
(self, x: SparseTensor)
sam3d_objects/model/backbone/tdfy_dit/modules/sparse/norm.py:63
Methodforward
(self, input: SparseTensor)
sam3d_objects/model/backbone/tdfy_dit/modules/sparse/nonlinearity.py:15
Methodforward
(self, input: SparseTensor)
sam3d_objects/model/backbone/tdfy_dit/modules/sparse/nonlinearity.py:20
Methodforward
(self, input: SparseTensor)
sam3d_objects/model/backbone/tdfy_dit/modules/sparse/nonlinearity.py:29
Methodforward
( self, x: Union[SparseTensor, torch.Tensor] )
sam3d_objects/model/backbone/tdfy_dit/modules/sparse/attention/modules.py:22
Methodforward
( self, x: Union[SparseTensor, torch.Tensor], context: Optional[Union[SparseTensor, to
sam3d_objects/model/backbone/tdfy_dit/modules/sparse/attention/modules.py:125
Methodforward
(self, x: SparseTensor)
sam3d_objects/model/backbone/tdfy_dit/modules/sparse/conv/conv_spconv.py:58
Methodforward
(self, x: SparseTensor)
sam3d_objects/model/backbone/tdfy_dit/modules/sparse/conv/conv_spconv.py:115
Methodforward
(self, x: SparseTensor)
sam3d_objects/model/backbone/tdfy_dit/modules/sparse/transformer/blocks.py:21
Methodforward
(self, x: SparseTensor)
sam3d_objects/model/backbone/tdfy_dit/modules/sparse/transformer/blocks.py:78
Methodforward
(self, x: SparseTensor, context: torch.Tensor)
sam3d_objects/model/backbone/tdfy_dit/modules/sparse/transformer/blocks.py:156
Methodforward
(self, x: SparseTensor, mod: torch.Tensor)
sam3d_objects/model/backbone/tdfy_dit/modules/sparse/transformer/modulated.py:81
Methodforward
( self, x: SparseTensor, mod: torch.Tensor, context: torch.Tensor )
sam3d_objects/model/backbone/tdfy_dit/modules/sparse/transformer/modulated.py:178
Methodforward
Args: x (torch.Tensor): (N, D) tensor of spatial positions
sam3d_objects/model/backbone/tdfy_dit/modules/transformer/blocks.py:37
Methodforward
(self, x: torch.Tensor)
sam3d_objects/model/backbone/tdfy_dit/modules/transformer/blocks.py:68
Methodforward
(self, x: torch.Tensor)
sam3d_objects/model/backbone/tdfy_dit/modules/transformer/blocks.py:119
Methodforward
(self, x: torch.Tensor, context: torch.Tensor)
sam3d_objects/model/backbone/tdfy_dit/modules/transformer/blocks.py:191
Methodforward
(self, x: torch.Tensor, mod: torch.Tensor)
sam3d_objects/model/backbone/tdfy_dit/modules/transformer/modulated.py:76
Methodforward
(self, x: torch.Tensor, mod: torch.Tensor, context: torch.Tensor)
sam3d_objects/model/backbone/tdfy_dit/modules/transformer/modulated.py:165
Methodforward
(self, x: Dict, mod: torch.Tensor, context: torch.Tensor)
sam3d_objects/model/backbone/tdfy_dit/modules/transformer/modulated.py:335
Methodforward
( self, x: NamedTuple, t: torch.Tensor, cond: torch.Tensor, )
sam3d_objects/model/backbone/tdfy_dit/models/sparse_structure_flow.py:213
Methodforward
( self, x: torch.Tensor, t: torch.Tensor, *condition_args, **condition
sam3d_objects/model/backbone/tdfy_dit/models/sparse_structure_flow.py:271
Methodforward
( self, latents_dict: dict, t: torch.Tensor, *condition_args, **condit
sam3d_objects/model/backbone/tdfy_dit/models/mot_sparse_structure_flow.py:205
Methodforward
(self, x: torch.Tensor)
sam3d_objects/model/backbone/tdfy_dit/models/sparse_structure_vae.py:81
Methodforward
(self, x: torch.Tensor)
sam3d_objects/model/backbone/tdfy_dit/models/sparse_structure_vae.py:108
Methodforward
( self, x: torch.Tensor, sample_posterior: bool = False, return_raw: bool = False )
sam3d_objects/model/backbone/tdfy_dit/models/sparse_structure_vae.py:199
Methodforward
(self, x: torch.Tensor)
sam3d_objects/model/backbone/tdfy_dit/models/sparse_structure_vae.py:310
Methodforward
(self, x: torch.Tensor)
sam3d_objects/model/backbone/tdfy_dit/models/sparse_structure_vae.py:391
Methodforward
(self, x: sp.SparseTensor, emb: torch.Tensor)
sam3d_objects/model/backbone/tdfy_dit/models/structured_latent_flow.py:61
Methodforward
( self, x: torch.Tensor, t: torch.Tensor, *condition_args, **condition
sam3d_objects/model/backbone/tdfy_dit/models/structured_latent_flow.py:319
Methodforward
(self, t)
sam3d_objects/model/backbone/tdfy_dit/models/timestep_embedder.py:54
Methodforward
(self, x: sp.SparseTensor)
sam3d_objects/model/backbone/tdfy_dit/models/structured_latent_vae/decoder_rf.py:124
Methodforward
(self, x: sp.SparseTensor)
sam3d_objects/model/backbone/tdfy_dit/models/structured_latent_vae/decoder_gs.py:152
Methodforward
(self, x: sp.SparseTensor)
sam3d_objects/model/backbone/tdfy_dit/models/structured_latent_vae/base.py:127
Methodforward
Apply the block to a Tensor, conditioned on a timestep embedding. Args: x: an [N x C x ...] Tensor of features.
sam3d_objects/model/backbone/tdfy_dit/models/structured_latent_vae/decoder_mesh.py:69
Methodforward
(self, x: sp.SparseTensor)
sam3d_objects/model/backbone/tdfy_dit/models/structured_latent_vae/decoder_mesh.py:183
Methodforward
(self, x: sp.SparseTensor, sample_posterior=True, return_raw=False)
sam3d_objects/model/backbone/tdfy_dit/models/structured_latent_vae/encoder.py:60
Methodforward
xyz : (B, 3, H, W) map of (x,y,z) coordinates valid_mask : (B, H, W) boolean - True for valid points (optional) retur
sam3d_objects/model/backbone/dit/embedder/pointmap.py:227
Methodforward
(self, *args, **kwargs)
sam3d_objects/model/backbone/dit/embedder/embedder_fuser.py:203
Methodforward
Apply remapping to point coordinates.
sam3d_objects/model/backbone/dit/embedder/point_remapper.py:20
Methodforward
(self, image, mask)
sam3d_objects/model/backbone/dit/embedder/dino.py:141
Methodforward
(self, x)
sam3d_objects/model/layers/llama3/ff.py:37
Methodfrom_features
(self, features)
sam3d_objects/model/backbone/tdfy_dit/representations/gaussian/gaussian_model.py:118
Methodfrom_instance_pose
(instance_pose: InstancePose)
sam3d_objects/data/dataset/tdfy/pose_target.py:204
Methodfrom_instance_pose
(cls, instance_pose: InstancePose)
sam3d_objects/data/dataset/tdfy/pose_target.py:273
Methodfrom_instance_pose
(cls, instance_pose: InstancePose, normalize: bool = False)
sam3d_objects/data/dataset/tdfy/pose_target.py:297
Methodfrom_instance_pose
(cls, instance_pose: InstancePose, normalize: bool = False)
sam3d_objects/data/dataset/tdfy/pose_target.py:397
Methodfrom_instance_pose
(cls, instance_pose: InstancePose, normalize: bool = False)
sam3d_objects/data/dataset/tdfy/pose_target.py:485
Methodfrom_invariant
(cls, invariant_targets: InvariantPoseTarget)
sam3d_objects/data/dataset/tdfy/pose_target.py:265
Methodfrom_invariant
(cls, invariant_targets: InvariantPoseTarget, normalize: bool = False)
sam3d_objects/data/dataset/tdfy/pose_target.py:356
Methodfrom_invariant
(cls, invariant_targets: InvariantPoseTarget)
sam3d_objects/data/dataset/tdfy/pose_target.py:454
Methodfrom_invariant
(cls, invariant_targets: InvariantPoseTarget, normalize: bool = False)
sam3d_objects/data/dataset/tdfy/pose_target.py:554
Methodfrom_invariant
(cls, invariant_targets: InvariantPoseTarget)
sam3d_objects/data/dataset/tdfy/pose_target.py:568
Methodfrom_invariant
(cls, invariant_targets: InvariantPoseTarget)
sam3d_objects/data/dataset/tdfy/pose_target.py:637
Methodfrom_invariant
(cls, invariant_targets: InvariantPoseTarget)
sam3d_objects/data/dataset/tdfy/pose_target.py:664
Methodfrom_invariant
(cls, invariant_targets: InvariantPoseTarget)
sam3d_objects/data/dataset/tdfy/pose_target.py:724
Methodfrom_opacity
(self, opacities)
sam3d_objects/model/backbone/tdfy_dit/representations/gaussian/gaussian_model.py:121
Functionfrom_pretrained
Load a model from a pretrained checkpoint. Args: path: The path to the checkpoint. Can be either local path or a Hugging Face model
sam3d_objects/model/backbone/tdfy_dit/models/__init__.py:8
Functionfull_instance_name
(instance: Any)
sam3d_objects/config/utils.py:60
Methodgenerate_iter
( self, x_shape, x_device, *args_conditionals, **kwargs_conditionals,
sam3d_objects/model/backbone/generator/flow_matching/model.py:202
Methodgenerate_iter
( self, x_shape, x_device, *args_conditionals, **kwargs_conditionals,
sam3d_objects/model/backbone/generator/flow_matching/model.py:331
Methodgenerate_iter
Generate samples using shortcut model
sam3d_objects/model/backbone/generator/shortcut/model.py:414
Functionget_attr
(data, key: str)
sam3d_objects/config/utils.py:173
Methodget_color
(self)
sam3d_objects/model/backbone/tdfy_dit/representations/octree/octree_dfs.py:241
Methodget_decoupoly
(self)
sam3d_objects/model/backbone/tdfy_dit/representations/octree/octree_dfs.py:237
Functionget_default_preprocessor
()
sam3d_objects/pipeline/preprocess_utils.py:19
Methodget_density
(self)
sam3d_objects/model/backbone/tdfy_dit/representations/octree/octree_dfs.py:218
Methodget_depth
(self)
sam3d_objects/model/backbone/tdfy_dit/representations/octree/octree_dfs.py:214
Functionget_expon_lr_func
Copied from Plenoxels Continuous learning rate decay function. Adapted from JaxNeRF The returned rate is lr_init when step=0 and lr_fina
sam3d_objects/model/backbone/tdfy_dit/representations/gaussian/general_utils.py:33
Methodget_features
(self)
sam3d_objects/model/backbone/tdfy_dit/representations/gaussian/gaussian_model.py:92
Methodget_features
(self)
sam3d_objects/model/backbone/tdfy_dit/representations/octree/octree_dfs.py:245
Functionget_img_color_augmentation
( color_jit_prob=0.5, gaussian_blur_prob=0.1, )
sam3d_objects/data/dataset/tdfy/img_processing.py:167
Functionget_item
(data, key: str)
sam3d_objects/config/utils.py:169
Functionget_last_checkpoint
(path: str)
sam3d_objects/model/io.py:117
Methodget_opacity
(self)
sam3d_objects/model/backbone/tdfy_dit/representations/gaussian/gaussian_model.py:100
Methodget_opacity
(self)
sam3d_objects/model/backbone/tdfy_dit/representations/octree/octree_dfs.py:229
Methodget_params
(img, scale, ratio)
sam3d_objects/data/dataset/tdfy/img_processing.py:22
Methodget_rotation
(self)
sam3d_objects/model/backbone/tdfy_dit/representations/gaussian/gaussian_model.py:84
Functionget_rotation_about_x_axis
(angle: float = math.pi / 2)
sam3d_objects/data/dataset/tdfy/transforms_3d.py:47
Methodget_scale_and_shift
(cls, pointmap)
sam3d_objects/data/dataset/tdfy/pose_target.py:459
Methodget_scaling
(self)
sam3d_objects/model/backbone/tdfy_dit/representations/gaussian/gaussian_model.py:77
Methodget_trivec
(self)
sam3d_objects/model/backbone/tdfy_dit/representations/octree/octree_dfs.py:233
Methodget_xyz
(self)
sam3d_objects/model/backbone/tdfy_dit/representations/gaussian/gaussian_model.py:88
Methodget_xyz
(self)
sam3d_objects/model/backbone/tdfy_dit/representations/octree/octree_dfs.py:210
Methodhalf
(self)
sam3d_objects/model/backbone/tdfy_dit/modules/sparse/basic.py:246
Functionhelper
(step)
sam3d_objects/model/backbone/tdfy_dit/representations/gaussian/general_utils.py:51
Methodinitialize_weights
(self)
sam3d_objects/model/backbone/tdfy_dit/models/structured_latent_vae/base.py:117
Functioninteractive_visualizer
(ply_path)
notebook/inference.py:407
Methodis_shape
(maybe_shape)
sam3d_objects/model/backbone/generator/flow_matching/model.py:100
Functionjson_to_halo_payloads
(target_data)
sam3d_objects/pipeline/inference_utils.py:879
Methodlayout
(self)
sam3d_objects/model/backbone/tdfy_dit/modules/sparse/basic.py:162
Functionlayout_post_optimization
( Mesh, quaternion, translation, scale, mask, point_map, intrinsics, Enable_sh
sam3d_objects/pipeline/inference_utils.py:83
Functionlayout_post_optimization_method_GS
Gaussian Splatting layout post-optimization function. Similar to layout_post_optimization but works with Gaussian Splatting instead of meshes
sam3d_objects/pipeline/inference_utils.py:239
Methodload_ply
(self, path)
sam3d_objects/model/backbone/tdfy_dit/representations/gaussian/gaussian_model.py:163
Functionload_rgb
Load a RGB(A) image from a file path.
sam3d_objects/data/dataset/tdfy/img_and_mask_transforms.py:97
Methodlog_likelihood
( self, x1, solver=None, steps=None, z_samples=1, *args_condit
sam3d_objects/model/backbone/generator/flow_matching/model.py:235
Functionlognorm_sampler
(mean=0.0, std=1.0, **kwargs)
sam3d_objects/model/backbone/generator/flow_matching/model.py:26
Methodloss
(self, x, *args_conditionals, **kwargs_conditionals)
sam3d_objects/model/backbone/generator/base.py:64
← previousnext →601–700 of 784, ranked by callers