MCPcopy Create free account

hub / github.com/OrangeSodahub/ORV / functions

Functions239 in github.com/OrangeSodahub/ORV

↓ 1 callersFunctionpostprocess_captions
(data_dir: str)
orv/dataset/prepare_dataset.py:1583
↓ 1 callersFunctionpostprocess_labels
(data_dir: str, splits: list[str])
orv/dataset/prepare_dataset.py:1469
↓ 1 callersMethodprepare_latents
Added support for 5D input images/latents
orv/models/cogvideox_control.py:1115
↓ 1 callersFunctionprepare_rotary_positional_embeddings
( height: int, width: int, num_frames: int, vae_scale_factor_spatial: int = 8, patch_size:
orv/utils.py:196
↓ 1 callersMethodpreprocess
Preprocess the image input. Args: image (`PipelineImageInput`): The image input, accepted formats are PI
orv/models/components.py:236
↓ 1 callersFunctionprocess_bridge
()
orv/dataset/data_process.py:419
↓ 1 callersFunctionprocess_droid
()
orv/dataset/data_process.py:346
↓ 1 callersFunctionprocess_nksr
(reconstructor, point_cloud_data, max_nn: int=20, device=torch.device('cuda'))
orv/dataset/prepare_dataset.py:727
↓ 1 callersFunctionproject_3d_to_2d
(points_3d, extrin, intrin)
orv/dataset/prepare_dataset.py:878
↓ 1 callersFunctionproject_3d_to_2d_np
(points_3d, extrin, intrin)
orv/dataset/prepare_dataset.py:1827
↓ 1 callersFunctionrasterize_gaussians
( means3D, means2D, sh, colors_precomp, language_feature_precomp, opacities, scale
orv/ops/diff-gaussian-rasterization/diff_gaussian_rasterization/__init__.py:21
↓ 1 callersFunctionread_mp4
(path: str)
orv/dataset/prepare_dataset.py:95
↓ 1 callersFunctionreconstruction_multi_task
(process_keys, rank: Optional[str] = None)
orv/dataset/prepare_dataset.py:2243
↓ 1 callersFunctionrender
Render the scene. Background tensor (bg_color) must be on GPU!
orv/dataset/gs_render.py:97
↓ 1 callersFunctionrmdir_recursive
(dir: pathlib.Path)
orv/dataset/encode_dataset.py:1055
↓ 1 callersFunctionrmdir_recursive
(dir: pathlib.Path)
orv/pipeline/evaluation_control_to_video.py:426
↓ 1 callersFunctionsave_depth
(depth: torch.Tensor, path: pathlib.Path)
orv/dataset/encode_dataset.py:300
↓ 1 callersFunctionserialize_artifacts
( fps: int, n_view: int, is_multiview: bool, filenames: list[str], images_dir: Optional[pa
orv/dataset/encode_dataset.py:320
↓ 1 callersFunctionserialize_artifacts
( gifs_folder: Path, mp4s_folder: Path, data_infos: dict, num_frames: int, num_views: int,
orv/pipeline/evaluation_control_to_video.py:27
↓ 1 callersFunctionserialize_artifacts_for_cascaded_videos
( gifs_folder: Path, mp4s_folder: Path, data_infos: dict, num_frames: int, num_views: int,
orv/pipeline/evaluation_control_to_video.py:66
↓ 1 callersFunctionto_tensor
Convert tensor data type from uint8 to float, divide value by 255.0 and permute the dimensions of clip tensor Args: clip (torch.t
orv/utils.py:263
↓ 1 callersFunctiontransformPoint4x3
orv/ops/diff-gaussian-rasterization/cuda_rasterizer/auxiliary.h:58
↓ 1 callersFunctiontransformPoint4x4
orv/ops/diff-gaussian-rasterization/cuda_rasterizer/auxiliary.h:68
↓ 1 callersFunctionupload
()
scripts/hf_model.py:12
MethodDeviceRegistry
orv/ops/include/pytorch_device_registry.hpp:49
FunctionDispatch
orv/ops/include/pytorch_device_registry.hpp:112
FunctionGET_BLOCKS
orv/ops/include/pytorch_cuda_helper.hpp:35
FunctionPYBIND11_MODULE
orv/ops/voxelize/voxelization_cpu.cpp:232
FunctionPYBIND11_MODULE
orv/ops/voxelize/voxelization.cpp:141
FunctionPYBIND11_MODULE
orv/ops/diff-gaussian-rasterization/ext.cpp:15
MethodRegister
orv/ops/include/pytorch_device_registry.hpp:35
Method__call__
Apply the transformation to each frame in the video. Args: video_frames (torch.Tensor): A tensor representing a batch of vide
orv/utils.py:288
Method__call__
Args: clip (torch.tensor, dtype=torch.uint8): Size is (T, C, H, W) Return: clip (torch.tensor, dtype=torch.fl
orv/utils.py:310
Method__call__
data keys: `action`, `latent`, `depths`, `semantics`, `metainfo` Returns: `actions`: [batch_size, n_frame - 1
orv/dataset/dataset.py:2058
Method__call__
( self, attn: Attention, hidden_states: torch.Tensor, encoder_hidden_states: O
orv/models/cogvideox_control.py:202
Method__call__
Args: image: [[[Image, Image], [Imgage, Image]], [[Image, Image], [Image, Image]], ...], n_batch -> n_view -> n_frame
orv/models/cogvideox_control.py:1228
Method__getitem__
( self, index_or_res: int | tuple, camera_index: Optional[int] = 0, return_vid
orv/dataset/dataset.py:1366
Method__getitem__
(self, index: int, camera_index: Optional[int] = 0, raise_error: bool = False)
orv/dataset/dataset.py:1642
Method__getitem__
(self, index_or_res: int | tuple)
orv/dataset/dataset.py:1802
Method__getitem__
(self, index)
orv/pipeline/compute_metrics.py:161
Method__init__
Initialize the preprocessing class with the target size. Args: size (tuple): The target height and width as a tuple (height,
orv/utils.py:280
Method__init__
(self)
orv/utils.py:307
Method__init__
(self, raster_settings)
orv/ops/diff-gaussian-rasterization/diff_gaussian_rasterization/__init__.py:185
Method__init__
( self, data_root: str, split: str = 'test', video_size: list[int] = [320, 480
orv/dataset/dataset.py:1526
Method__init__
( self, n_view: int, camera_ids: list[str | int], **kwargs, )
orv/dataset/dataset.py:1717
Method__init__
( self, data_source: RobotDataset | MultiViewRobotDataset, batch_size: int = 8,
orv/dataset/dataset.py:1990
Method__init__
(self, weight_dtype: torch.dtype, load_tensors: bool)
orv/dataset/dataset.py:2054
Method__init__
( self, conditioning_dim: int, embedding_dim: int, elementwise_affine: bool =
orv/models/cogvideox_control.py:43
Method__init__
( self, dim: int, num_attention_heads: int, attention_head_dim: int, t
orv/models/cogvideox_control.py:353
Method__init__
( self, num_attention_heads: int = 30, attention_head_dim: int = 64, in_channe
orv/models/cogvideox_control.py:453
Method__init__
( self, tokenizer: T5Tokenizer, text_encoder: T5EncoderModel, vae: Autoencoder
orv/models/cogvideox_control.py:1094
Method__init__
( self, state_dim: int, hidden_size: int, dropout: float = 0., compres
orv/models/components.py:22
Method__init__
( self, state_dim: int, hidden_size: int, compress_ratio: int = 1, )
orv/models/components.py:76
Method__init__
( self, conditioning_embedding_channels: int, conditioning_channels: int = 3,
orv/models/components.py:167
Method__init__
(self, files, num_frames, transforms=None)
orv/pipeline/compute_metrics.py:153
Method__iter__
(self)
orv/dataset/dataset.py:2016
Method__len__
(self)
orv/dataset/dataset.py:595
Method__len__
(self)
orv/dataset/dataset.py:2008
Method__len__
(self)
orv/pipeline/compute_metrics.py:158
Method__repr__
(self)
orv/utils.py:319
Method_check_render_data
(self, render_file)
orv/dataset/dataset.py:428
Method_fetch_episode_ids
(ann_file, valid_ids)
orv/dataset/dataset.py:401
Function_handle_terminate
(signum, frame)
orv/dataset/prepare_dataset.py:495
Method_load_and_process_ann_file
(self, i)
orv/dataset/dataset.py:466
Method_load_and_process_ann_file
(self, i)
orv/dataset/dataset.py:1234
Method_load_and_process_ann_file
(self, i)
orv/dataset/dataset.py:1570
Function_load_episode_bridgev2
(episode: dict)
orv/dataset/data_process.py:150
Function_load_episode_droid
(episode: dict)
orv/dataset/data_process.py:20
Method_load_video
(self, video_path, frame_ids)
orv/dataset/dataset.py:600
Function_postprocess_labels
(load_dir: str, mask_annotator: sv.MaskAnnotator, colors60_list: list[tuple[int]], data: tuple)
orv/dataset/prepare_dataset.py:1377
Functionapply_depth_colormap
Input: gray: gray image, tensor/numpy, (H, W) Output: depth: (3, H, W), tensor
orv/dataset/prepare_dataset.py:1803
Functionapply_semantic_colormap
Input: semantic: semantic image, tensor/numpy, (N, H, W) Output: depth: (3, H, W), tensor
orv/dataset/gs_render.py:37
Functioncheck_frames
(x: Any)
orv/dataset/encode_dataset.py:67
Functioncheck_height
(x: Any)
orv/dataset/encode_dataset.py:49
Functioncheck_width
(x: Any)
orv/dataset/encode_dataset.py:58
Functioncollate_fn
(data)
orv/dataset/encode_dataset.py:648
Functioncompute_scale_and_shift
(prediction, target, mask)
orv/dataset/prepare_dataset.py:1858
Methodcreate_custom_forward
(module)
orv/models/cogvideox_control.py:750
Methodcustom_forward
(*inputs)
orv/models/cogvideox_control.py:751
Functiondepths_to_points
Convert depth map to 3D points using camera intrinsics. Args: depth_map (npt.NDArray): Depth map of shape (H, W) containing depth va
orv/dataset/prepare_dataset.py:102
Functiondnormvdv
orv/ops/diff-gaussian-rasterization/cuda_rasterizer/auxiliary.h:107
Functiondnormvdz
orv/ops/diff-gaussian-rasterization/cuda_rasterizer/auxiliary.h:99
Functiondynamic_voxelize_forward
orv/ops/voxelize/voxelization_cpu.cpp:215
Functiondynamic_voxelize_forward
orv/ops/voxelize/voxelization.cpp:124
Functiondynamic_voxelize_forward_cpu
orv/ops/voxelize/voxelization_cpu.cpp:107
Functiondynamic_voxelize_forward_cpu_kernel
orv/ops/voxelize/voxelization_cpu.cpp:7
Functiondynamic_voxelize_forward_cuda
orv/ops/voxelize/voxelization.cpp:48
Functionencode_empty_prompt
()
orv/dataset/encode_dataset.py:1069
Functionepisode_loader
(queue, load_func)
orv/dataset/data_process.py:328
Functionfind_largest_label_id
(data_dir)
orv/dataset/prepare_dataset.py:2379
Methodforward
Convert points(N, >=3) to voxels. Args: points (torch.Tensor): [N, ndim]. Points[:, :3] contain xyz points an
orv/ops/voxelize/voxelization.py:45
Methodforward
( ctx, means3D, means2D, sh, colors_precomp, language_feature_
orv/ops/diff-gaussian-rasterization/diff_gaussian_rasterization/__init__.py:48
Methodforward
(self, means3D, means2D, opacities, shs = None, colors_precomp = None, language_feature_precomp=None, scales =
orv/ops/diff-gaussian-rasterization/diff_gaussian_rasterization/__init__.py:200
Methodforward
( self, hidden_states: torch.Tensor, encoder_hidden_states: torch.Tensor, temb
orv/models/cogvideox_control.py:60
Methodforward
( self, x: torch.Tensor, timestep: Optional[torch.Tensor] = None, temb: Option
orv/models/cogvideox_control.py:155
Methodforward
( self, hidden_states: torch.Tensor, encoder_hidden_states: torch.Tensor, temb
orv/models/cogvideox_control.py:313
Methodforward
( self, hidden_states: torch.Tensor, encoder_hidden_states: torch.Tensor, temb
orv/models/cogvideox_control.py:394
Methodforward
( self, hidden_states: torch.Tensor, encoder_hidden_states: torch.Tensor, cont
orv/models/cogvideox_control.py:715
Methodforward
(self, x)
orv/models/components.py:47
Methodforward
(self, x)
orv/models/components.py:92
← previousnext →101–200 of 239, ranked by callers