MCPcopy Create free account

hub / github.com/VAST-AI-Research/MIDI-3D / functions

Functions380 in github.com/VAST-AI-Research/MIDI-3D

⨍ Functions 380◇ Types & classes 47↳ Endpoints 1
Methodon_predict_batch_end
(self, outputs, batch, batch_idx)
midi/systems/base.py:239
Methodon_predict_batch_start
(self, batch, batch_idx, dataloader_idx=0)
midi/systems/base.py:206
Methodon_predict_end
(self)
midi/systems/base.py:177
Methodon_predict_epoch_end
(self)
midi/systems/base.py:174
Methodon_predict_start
(self, trainer, pl_module)
midi/utils/callbacks.py:157
Methodon_predict_start
(self)
midi/systems/base.py:168
Methodon_test_batch_end
(self, outputs, batch, batch_idx)
midi/systems/base.py:229
Methodon_test_batch_start
(self, batch, batch_idx, dataloader_idx=0)
midi/systems/base.py:200
Methodon_test_end
(self)
midi/systems/system_midi.py:403
Methodon_test_end
(self)
midi/systems/base.py:164
Methodon_test_epoch_end
(self)
midi/systems/base.py:161
Methodon_test_start
(self, trainer, pl_module)
midi/utils/callbacks.py:153
Methodon_train_batch_end
(self, trainer, pl_module, *args, **kwargs)
midi/utils/callbacks.py:143
Methodon_train_batch_end
(self, outputs, batch, batch_idx)
midi/systems/system_midi.py:149
Methodon_train_batch_end
(self, outputs, batch, batch_idx)
midi/systems/base.py:212
Methodon_train_batch_start
(self, batch, batch_idx, unused=0)
midi/systems/base.py:188
Methodon_train_start
(self)
midi/systems/system_midi.py:134
Methodon_validation_batch_end
(self, outputs, batch, batch_idx)
midi/systems/base.py:219
Methodon_validation_batch_start
(self, batch, batch_idx, dataloader_idx=0)
midi/systems/base.py:194
Methodon_validation_epoch_end
(self)
midi/systems/system_midi.py:348
Methodon_validation_epoch_end
(self)
midi/systems/base.py:155
Methodon_validation_epoch_start
(self)
midi/systems/system_midi.py:345
Methodon_validation_start
(self, trainer, pl_module)
midi/utils/callbacks.py:149
Functionparse_scheduler_to_instance
(config, optimizer)
midi/utils/training_utils.py:52
Functionpolar_to_c2w
Compute L = p - C. Normalize L. Compute s = L x u. (cross product) Normalize s. Compute u' = s x L. rotation = [s, u, -l]
midi/utils/ops.py:411
Methodpredict_dataloader
(self)
midi/data/multi_object.py:677
Methodpredict_step
(self, batch, batch_idx)
midi/systems/base.py:171
Methodprepare_data
(self)
midi/data/multi_object.py:647
Functionpreprocess_image_for_clip
( image: Float[Tensor, "B C H W"], do_resize=True, size: Optional[int] = None )
midi/systems/model_utils.py:25
Functionpreprocess_image_for_dinov2
( image: Float[Tensor, "B C H W"], do_resize=True, size: Optional[int] = None )
midi/systems/model_utils.py:35
Methodquery_fn
(q, kv)
midi/models/autoencoders/autoencoder_kl_triposg.py:202
Functionrays_intersect_bbox
( rays_o: Float[Tensor, "N 3"], rays_d: Float[Tensor, "N 3"], radius: Float, near: Float = 0.0
midi/utils/ops.py:346
Functionread_mesh_from_path
(mesh_path)
midi/utils/mesh_process.py:11
Functionreflect
(x, n)
midi/utils/ops.py:19
Methodregister_non_module
(self, name: str, module: nn.Module)
midi/utils/base.py:118
Methodresumed
(self)
midi/systems/base.py:87
Functionrun_generation
( rgb_image: Any, seg_image: Union[str, Image.Image], seed: int, randomize_seed: bool = False,
gradio_demo.py:190
Functionrun_segmentation
( image_prompts: Any, seg_mode: str, text_labels: Optional[str] = None, polygon_refinement: bo
gradio_demo.py:136
Methodsave_cubemap
(self, filename, img, data_range=(0, 1), rgba=False)
midi/utils/saving.py:318
Methodsave_custom_adapter
( self, save_directory: Union[str, os.PathLike], weight_name: str, safe_serial
midi/loaders/custom_adapter.py:68
Methodsave_data
(self, filename, data)
midi/utils/saving.py:353
Methodsave_function
(weights, filename)
midi/loaders/custom_adapter.py:83
Methodsave_grayscale_image
( self, filename, img, data_range=DEFAULT_GRAYSCALE_KWARGS["data_range"],
midi/utils/saving.py:226
Methodsave_img_sequences
( self, seq_dir, matcher, save_format="mp4", fps=30, delete=Tr
midi/utils/saving.py:405
Methodsave_json
(self, filename, payload)
midi/utils/saving.py:442
Methodsave_rgb_image
( self, filename, img, data_format=DEFAULT_RGB_KWARGS["data_format"],
midi/utils/saving.py:117
Methodsave_state_dict
(self, filename, data)
midi/utils/saving.py:367
Methodsave_uv_image
( self, filename, img, data_format=DEFAULT_UV_KWARGS["data_format"], d
midi/utils/saving.py:153
Methodsavedir
(self)
midi/utils/callbacks.py:49
Methodscale_noise
Forward function for the noise scaling in the flow matching.
midi/schedulers/scheduling_rectified_flow.py:310
Functionscale_tensor
( dat: Num[Tensor, "... D"], inp_scale: ValidScale, tgt_scale: ValidScale )
midi/utils/ops.py:26
Methodset_begin_index
Sets the begin index for the scheduler. This function should be run from pipeline before the inference. Args: begin_inde
midi/schedulers/scheduling_rectified_flow.py:144
Methodset_default_attn_processor
Disables custom attention processors and sets the default attention implementation.
midi/models/transformers/triposg_transformer.py:608
Methodset_default_attn_processor
Disables custom attention processors and sets the default attention implementation.
midi/models/autoencoders/autoencoder_kl_triposg.py:374
Functionshow_vram_usage
(name)
midi/utils/misc.py:296
Functionsigned_distance_function
Return the distance to the 0.5 levelset of a function, the mask of the border (i.e., the nearest cells to the 0.5 level-set) and the mask of
midi/utils/smoothing.py:315
Functionsigned_distance_function_iso0
Return the distance to the 0 levelset of a function, the mask of the border (i.e., the nearest cells to the 0 level-set) and the mask of the
midi/utils/smoothing.py:338
Functionsmooth
Smooths the 0.5 level-set of a binary array. Returns a floating-point array with a smoothed version of the original level-set in the 0 isoval
midi/utils/smoothing.py:574
Methodstart
(self, name: str)
midi/utils/misc.py:253
Methodstep_index
The index counter for current timestep. It will increase 1 after each scheduler step.
midi/schedulers/scheduling_rectified_flow.py:130
Methodtest_step
(self, batch, batch_idx)
midi/systems/system_midi.py:351
Methodtest_step
(self, batch, batch_idx)
midi/systems/base.py:158
Functiontet_sdf_diff
( vert_sdf: Float[Tensor, "Nv 1"], tet_edges: Integer[Tensor, "Ne 2"] )
midi/utils/ops.py:323
Functiontime_recorder_enabled
()
midi/utils/misc.py:287
Methodtrain_dataloader
(self)
midi/data/multi_object.py:650
Methodtraining_step
Training step for the pl module. Args: batch: The batch of data. Each batch data includes keys `num_instances`, `surface
midi/systems/system_midi.py:137
Methodtraining_step
(self, batch, batch_idx)
midi/systems/base.py:138
Methodtrue_current_epoch
(self)
midi/systems/base.py:99
Methodtrue_global_step
(self)
midi/systems/base.py:92
Methodunfuse_qkv_projections
Disable QKV projection fusion if enabled. <Tip warning={true}> This API is 🧪 experimental. </Tip> Args:
midi/pipelines/pipeline_utils.py:66
Methodunfuse_qkv_projections
Disables the fused QKV projection if enabled. <Tip warning={true}> This API is 🧪 experimental. </Tip>
midi/models/transformers/triposg_transformer.py:529
Methodunfuse_qkv_projections
Disables the fused QKV projection if enabled. <Tip warning={true}> This API is 🧪 experimental. </Tip>
midi/models/autoencoders/autoencoder_kl_triposg.py:295
Methodupdate_step
(self, epoch: int, global_step: int, on_load_weights: bool = False)
midi/systems/base.py:249
Methodval_dataloader
(self)
midi/data/multi_object.py:659
Functionvalidate_empty_rays
(ray_indices, t_start, t_end)
midi/utils/ops.py:337
Methodvalidation_step
(self, batch, batch_idx)
midi/systems/system_midi.py:342
Methodvalidation_step
(self, batch, batch_idx)
midi/systems/base.py:152
Methodversion
Get the experiment version. Returns: The experiment version if specified else the next version.
midi/utils/callbacks.py:26
Functionwrapper
(*args, **kwargs)
midi/utils/misc.py:202
Methodxyxy
(self)
scripts/grounding_sam.py:110
← previous301–380 of 380, ranked by callers