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
Method__rtruediv__
( self, other: Union[torch.Tensor, "SparseTensor", float] )
sam3d_objects/model/backbone/tdfy_dit/modules/sparse/basic.py:401
Method__sub__
( self, other: Union[torch.Tensor, "SparseTensor", float] )
sam3d_objects/model/backbone/tdfy_dit/modules/sparse/basic.py:376
Method__truediv__
( self, other: Union[torch.Tensor, "SparseTensor", float] )
sam3d_objects/model/backbone/tdfy_dit/modules/sparse/basic.py:396
Method_apply_add
(self, x, h)
sam3d_objects/model/backbone/tdfy_dit/modules/transformer/modulated.py:273
Method_apply_cross_attn
(self, h, cross_attn, context)
sam3d_objects/model/backbone/tdfy_dit/modules/transformer/modulated.py:264
Method_apply_mlp
(self, h, scale_mlp, shift_mlp)
sam3d_objects/model/backbone/tdfy_dit/modules/transformer/modulated.py:270
Method_apply_module
(self, x, module)
sam3d_objects/model/backbone/tdfy_dit/modules/attention/modules.py:271
Method_apply_module
(self, h, module)
sam3d_objects/model/backbone/tdfy_dit/modules/transformer/modulated.py:261
Method_apply_msa
(self, h, scale_msa, shift_msa)
sam3d_objects/model/backbone/tdfy_dit/modules/transformer/modulated.py:267
Method_apply_multiplication
(self, h, multiplier)
sam3d_objects/model/backbone/tdfy_dit/modules/transformer/modulated.py:276
Method_basic_init
(module)
sam3d_objects/model/backbone/tdfy_dit/models/mm_latent.py:34
Method_basic_init
(module)
sam3d_objects/model/backbone/tdfy_dit/models/sparse_structure_flow.py:187
Method_basic_init
(module)
sam3d_objects/model/backbone/tdfy_dit/models/mot_sparse_structure_flow.py:111
Method_basic_init
(module)
sam3d_objects/model/backbone/tdfy_dit/models/structured_latent_flow.py:238
Method_basic_init
(module)
sam3d_objects/model/backbone/tdfy_dit/models/structured_latent_vae/base.py:119
Method_cast_type
(self, x, dtype)
sam3d_objects/model/backbone/tdfy_dit/models/mot_sparse_structure_flow.py:137
Method_cfg_step_tensor
(self, y_cond, y_uncond, strength)
sam3d_objects/model/backbone/generator/classifier_free_guidance.py:99
Method_cfg_step_tensor
(self, y_cond, y_uncond, y_unpm, strength, strength_pm)
sam3d_objects/model/backbone/generator/classifier_free_guidance.py:168
Method_compile
(self)
sam3d_objects/pipeline/inference_pipeline_pointmap.py:102
Method_forward_intermediate_layers
( self, input_img, intermediate_layers, cls_token=True )
sam3d_objects/model/backbone/dit/embedder/dino.py:86
Method_generate_dynamics
( self, x_t, t, *args_conditionals, **kwargs_conditionals, )
sam3d_objects/model/backbone/generator/flow_matching/model.py:221
Method_generate_dynamics
Generate dynamics for ODE solver
sam3d_objects/model/backbone/generator/shortcut/model.py:435
Method_generate_t_and_d
Generate t and d together according to shortcut models paper. According to the paper: "During training, we first sample d, t
sam3d_objects/model/backbone/generator/shortcut/model.py:246
Method_generate_target_tensor
(self, x0: torch.Tensor, x1: torch.Tensor)
sam3d_objects/model/backbone/generator/flow_matching/model.py:123
Method_generate_x0_tensor
(self, x1: torch.Tensor)
sam3d_objects/model/backbone/generator/flow_matching/model.py:112
Method_generate_xt_tensor
(self, x0: torch.Tensor, x1: torch.Tensor, t: torch.Tensor)
sam3d_objects/model/backbone/generator/flow_matching/model.py:116
Method_get_shape
(self, x)
sam3d_objects/model/backbone/tdfy_dit/modules/attention/modules.py:288
Method_log_likelihood_dynamics
( self, state, t, device, z_samples, *args_conditionals,
sam3d_objects/model/backbone/generator/flow_matching/model.py:262
Method_preprocess_image_and_mask_inference
(self, rgb_image, mask_image)
sam3d_objects/data/dataset/tdfy/preprocessor.py:191
Method_preprocess_image_and_mask_pointmap
( self, rgb_image, mask_image, pointmap, img_mask_pointmap_joint_transform )
sam3d_objects/pipeline/inference_pipeline_pointmap.py:164
Method_process_image_and_mask_mess
Original method - calls extended version without pointmap
sam3d_objects/data/dataset/tdfy/preprocessor.py:144
Method_reshape
(self, qkv, tensor_shape, num_heads)
sam3d_objects/model/backbone/tdfy_dit/modules/attention/modules.py:263
Method_reshape_back
(self, qkv, tensor_shape)
sam3d_objects/model/backbone/tdfy_dit/modules/attention/modules.py:267
Method_solve_vd_QEF
(self, p_bxnx3, norm_bxnx3, c_bx3, qef_reg_scale)
sam3d_objects/model/backbone/tdfy_dit/representations/mesh/flexicubes/flexicubes.py:237
Functionadd_flag
(args, kwargs)
sam3d_objects/model/backbone/generator/classifier_free_guidance.py:53
Methodadd_noise
(mean_val)
sam3d_objects/model/backbone/generator/flow_matching/solver.py:76
Functionadd_prefix_state_dict_fn
(prefix: str)
sam3d_objects/model/io.py:97
Functionapply_blur_augmentation
Apply blur augmentation with configurable parameters
sam3d_objects/data/dataset/tdfy/img_and_mask_transforms.py:954
Functionbuild_batch_extractor
(mapping: MappingType)
sam3d_objects/data/utils.py:149
Methodbuild_covariance_from_scaling_rotation
(scaling, scaling_modifier, rotation)
sam3d_objects/model/backbone/tdfy_dit/representations/gaussian/gaussian_model.py:70
Methodclone_output_wrapper
(f)
sam3d_objects/pipeline/inference_pipeline.py:214
Functioncompiled_fn_wrapper
(*args, **kwargs)
sam3d_objects/pipeline/inference_pipeline_pointmap.py:78
Methodcomput_v_normals
(self, verts, faces)
sam3d_objects/model/backbone/tdfy_dit/representations/mesh/cube2mesh.py:36
Functioncompute_gs_loss
Compute loss for GS render-and-compare optimization. Similar to compute_loss but for GS.
sam3d_objects/pipeline/layout_post_optimization_utils.py:1118
Methodcompute_mean
(x, v)
sam3d_objects/model/backbone/generator/flow_matching/solver.py:68
Methodcompute_velocity
(k1, k2, k3, k4)
sam3d_objects/model/backbone/generator/flow_matching/solver.py:121
Functionconvert_module_to_f16
Convert primitive modules to float16.
sam3d_objects/model/backbone/tdfy_dit/modules/utils.py:34
Functionconvert_module_to_f32
Convert primitive modules to float32, undoing convert_module_to_f16().
sam3d_objects/model/backbone/tdfy_dit/modules/utils.py:44
Methodconvert_to_fp32
Convert the torso of the model to float32.
sam3d_objects/model/backbone/tdfy_dit/models/sparse_structure_flow.py:179
Methodconvert_to_fp32
Convert the torso of the model to float32.
sam3d_objects/model/backbone/tdfy_dit/models/mot_sparse_structure_flow.py:103
Methodconvert_to_fp32
Convert the torso of the model to float32.
sam3d_objects/model/backbone/tdfy_dit/models/sparse_structure_vae.py:190
Methodconvert_to_fp32
Convert the torso of the model to float32.
sam3d_objects/model/backbone/tdfy_dit/models/sparse_structure_vae.py:301
Methodconvert_to_fp32
Convert the torso of the model to float32.
sam3d_objects/model/backbone/tdfy_dit/models/structured_latent_flow.py:228
Methodconvert_to_fp32
Convert the torso of the model to float32.
sam3d_objects/model/backbone/tdfy_dit/models/structured_latent_vae/base.py:111
Methodconvert_to_fp32
Convert the torso of the model to float32.
sam3d_objects/model/backbone/tdfy_dit/models/structured_latent_vae/decoder_mesh.py:160
Methodcoords
(self)
sam3d_objects/model/backbone/tdfy_dit/modules/sparse/basic.py:180
Functioncrop_and_pad
Crop an image using a bounding box and pad with zeros if out of bounds. Args: image (torch.Tensor): CxHxW image. bbox (tuple
sam3d_objects/data/dataset/tdfy/img_and_mask_transforms.py:392
Functioncrop_around_mask_with_random_box_size_factor
( loaded_image: torch.Tensor, mask: torch.Tensor, random_box_size_factor: float = 1.0, pointma
sam3d_objects/data/dataset/tdfy/img_and_mask_transforms.py:247
Functioncrop_img_to_obj
(mask, context_size)
sam3d_objects/data/dataset/tdfy/img_processing.py:139
Methodcur_depth
(self)
sam3d_objects/model/backbone/tdfy_dit/representations/octree/octree_dfs.py:202
Functiondecode
(model_output_dict, scene_scale=None, scene_shift=None)
sam3d_objects/pipeline/inference_utils.py:468
Methoddenormalize
(self, pointmap: torch.Tensor, scale: torch.Tensor, shift: torch.Tensor)
sam3d_objects/data/dataset/tdfy/img_and_mask_transforms.py:513
Methoddenormalize
(self, pointmap: torch.Tensor, scale: torch.Tensor, shift: torch.Tensor)
sam3d_objects/data/dataset/tdfy/img_and_mask_transforms.py:714
Methoddenormalize
(self, value: torch.Tensor)
sam3d_objects/data/dataset/tdfy/pose_target.py:783
Methoddense
(self)
sam3d_objects/model/backbone/tdfy_dit/modules/sparse/basic.py:259
Methoddevice
Return the device of the model.
sam3d_objects/model/backbone/tdfy_dit/models/sparse_structure_flow.py:167
Methoddevice
Return the device of the model.
sam3d_objects/model/backbone/tdfy_dit/models/mot_sparse_structure_flow.py:91
Methoddevice
Return the device of the model.
sam3d_objects/model/backbone/tdfy_dit/models/sparse_structure_vae.py:175
Methoddevice
Return the device of the model.
sam3d_objects/model/backbone/tdfy_dit/models/sparse_structure_vae.py:286
Methoddevice
Return the device of the model.
sam3d_objects/model/backbone/tdfy_dit/models/structured_latent_flow.py:214
Methoddevice
Return the device of the model.
sam3d_objects/model/backbone/tdfy_dit/models/structured_latent_vae/base.py:99
Methoddicts_instance_pose_to_pose_target
( pose_target_convention: str, **kwargs, )
sam3d_objects/data/dataset/tdfy/pose_target.py:749
Functiondiscard
(args, kwargs)
sam3d_objects/model/backbone/generator/classifier_free_guidance.py:25
Functiondisplay_image
(image, masks=None)
notebook/inference.py:381
Functiondrop_tensors
(args, kwargs)
sam3d_objects/model/backbone/generator/classifier_free_guidance.py:47
Methoddtype
(self)
sam3d_objects/model/backbone/tdfy_dit/modules/sparse/basic.py:194
Functiondump_config
(config: DictConfig, path: str = "./config.yaml")
sam3d_objects/config/utils.py:12
Functionembedder
()
sam3d_objects/model/backbone/tdfy_dit/models/mm_latent.py:61
Functionexp_anealing
(optimizer, step, total_steps, start_lr, end_lr)
sam3d_objects/model/backbone/tdfy_dit/utils/postprocessing_utils.py:503
Functionexpand_as_right
(tensor, target)
sam3d_objects/data/utils.py:98
Functionexport_transformed_mesh_glb
( verts, mesh_obj, center, quat, translation, scale, output_path )
sam3d_objects/pipeline/layout_post_optimization_utils.py:281
Methodextra_repr
(self)
sam3d_objects/model/backbone/dit/embedder/point_remapper.py:77
Functionextract_fn
(batch)
sam3d_objects/data/utils.py:129
Methodfeats
(self)
sam3d_objects/model/backbone/tdfy_dit/modules/sparse/basic.py:166
Functionfind_and_ensure_is_subclass
(target: TargetType, type_: type)
sam3d_objects/config/utils.py:92
Methodflush
(self)
sam3d_objects/model/backbone/tdfy_dit/representations/gaussian/general_utils.py:144
Functionformat_to_halo
(layout_output)
sam3d_objects/pipeline/inference_utils.py:857
Methodforward
(self, x, t, *args_cond, **kwargs_cond)
sam3d_objects/model/backbone/generator/classifier_free_guidance.py:120
Methodforward
(self, x, t, *args_cond, **kwargs_cond)
sam3d_objects/model/backbone/generator/classifier_free_guidance.py:198
Methodforward
(self, x, t, *args_cond, p_unconditional=None, **kwargs_cond)
sam3d_objects/model/backbone/generator/classifier_free_guidance.py:230
Methodforward
(self, x: torch.Tensor)
sam3d_objects/model/backbone/tdfy_dit/modules/norm.py:16
Methodforward
(self, x: torch.Tensor)
sam3d_objects/model/backbone/tdfy_dit/modules/norm.py:21
Methodforward
(self, x: torch.Tensor)
sam3d_objects/model/backbone/tdfy_dit/modules/attention/modules.py:20
Methodforward
Args: q (sp.SparseTensor): [..., N, D] tensor of queries k (sp.SparseTensor): [..., N, D] tensor of keys
sam3d_objects/model/backbone/tdfy_dit/modules/attention/modules.py:48
Methodforward
( self, x: torch.Tensor, context: Optional[torch.Tensor] = None, indices: Opti
sam3d_objects/model/backbone/tdfy_dit/modules/attention/modules.py:138
Methodforward
( self, x: Dict, )
sam3d_objects/model/backbone/tdfy_dit/modules/attention/modules.py:339
Methodforward
(self, input: SparseTensor)
sam3d_objects/model/backbone/tdfy_dit/modules/sparse/linear.py:13
Methodforward
(self, input: SparseTensor)
sam3d_objects/model/backbone/tdfy_dit/modules/sparse/spatial.py:20
Methodforward
(self, input: SparseTensor)
sam3d_objects/model/backbone/tdfy_dit/modules/sparse/spatial.py:78
Methodforward
(self, input: SparseTensor)
sam3d_objects/model/backbone/tdfy_dit/modules/sparse/spatial.py:108
← previousnext →501–600 of 784, ranked by callers