MCPcopy Create free account

hub / github.com/OrangeSodahub/ORV / functions

Functions239 in github.com/OrangeSodahub/ORV

⨍ Functions 239◇ Types & classes 33
↓ 29 callersMethodfrom_pretrained
(cls, pretrained_model_name_or_path: Optional[Union[str, os.PathLike]], **kwargs)
orv/models/cogvideox_control.py:951
↓ 13 callersFunctionunwrap_model
(model)
orv/pipeline/train_cogvideox_control_to_video_sft.py:390
↓ 4 callersMethod__init__
( self, dim: int, num_attention_heads: int, attention_head_dim: int, t
orv/models/cogvideox_control.py:275
↓ 4 callersMethod_get_actions
(self, arm_states, gripper_states)
orv/dataset/dataset.py:960
↓ 4 callersMethod_get_frames
( self, frame_ids: list[int], video_path: Optional[str] = None, latent_video_p
orv/dataset/dataset.py:655
↓ 4 callersMethod_get_robot_states
(self, label, frame_ids)
orv/dataset/dataset.py:927
↓ 4 callersFunctioneuler2rotm
Euler angle (ZYX) to rotation matrix.
orv/dataset/dataset.py:65
↓ 3 callersFunctionGetDeviceStr
orv/ops/include/pytorch_device_registry.hpp:15
↓ 3 callersMethod__init__
(self, action_in_channel=128, out_channels=1152)
orv/models/components.py:122
↓ 3 callersMethod_get_cond_frames
( self, frame_ids: list[int], recon_file_path: Optional[str] = None, label_fil
orv/dataset/dataset.py:785
↓ 3 callersFunction_preprocess_episode
(episode, index, split, save_path)
orv/dataset/data_process.py:350
↓ 3 callersFunctiongenerate_colors
(n=60)
orv/dataset/prepare_dataset.py:1458
↓ 3 callersFunctionprint_memory
(device: Union[str, torch.device])
orv/utils.py:249
↓ 3 callersFunctionread_video_decord
(path)
orv/pipeline/compute_metrics.py:21
↓ 3 callersFunctionrun_validation
(step, epoch)
orv/pipeline/train_cogvideox_control_to_video_sft.py:635
↓ 2 callersMethod__init__
( self, # disk information data_root: str, split: str = 'train', rende
orv/dataset/dataset.py:160
↓ 2 callersFunction_compute_fid_stats_of_path
(files, model, batch_size, dims, device, num_frames, cache_path=None)
orv/pipeline/compute_metrics.py:209
↓ 2 callersFunction_get_fvd_feats
(videos, model, device)
orv/pipeline/compute_metrics.py:334
↓ 2 callersMethod_init_sequences
(self)
orv/dataset/dataset.py:443
↓ 2 callersFunction_preprocess
(files)
orv/pipeline/compute_metrics.py:309
↓ 2 callersMethod_set_trainable_parameters
(self)
orv/models/cogvideox_control.py:641
↓ 2 callersFunctionapply_depth_colormap
Input: gray: gray image, tensor/numpy, (H, W) Output: depth: (3, H, W), tensor
orv/dataset/gs_render.py:11
↓ 2 callersFunctioncompute_prompt_embeddings
( tokenizer: T5Tokenizer, text_encoder: T5EncoderModel, prompts: List[str], max_sequence_lengt
orv/dataset/encode_dataset.py:249
↓ 2 callersFunctioncpu_deep_copy_tuple
(input_tuple)
orv/ops/diff-gaussian-rasterization/diff_gaussian_rasterization/__init__.py:17
↓ 2 callersFunctiondynamic_voxelize_forward_impl
orv/ops/voxelize/voxelization.cpp:79
↓ 2 callersFunctionencode_prompt
( tokenizer: T5Tokenizer, text_encoder: T5EncoderModel, prompt: Union[str, List[str]], num_vid
orv/dataset/encode_dataset.py:225
↓ 2 callersFunctionencode_prompt
( tokenizer: T5Tokenizer, text_encoder: T5EncoderModel, prompt: Union[str, List[str]], num_vid
orv/models/text_encoder.py:45
↓ 2 callersFunctionfocal2fov
(focal, pixels)
orv/dataset/gs_render.py:174
↓ 2 callersFunctionget_args
()
orv/dataset/encode_dataset.py:76
↓ 2 callersFunctionget_gradient_norm
(parameters)
orv/utils.py:166
↓ 2 callersFunctionhard_voxelize_forward_impl
orv/ops/voxelize/voxelization.cpp:56
↓ 2 callersFunctionload_images
Open and convert all images or videos in a list or folder to proper input format for DUSt3R.
orv/dataset/prepare_dataset.py:279
↓ 2 callersFunctionreset_memory
(device: Union[str, torch.device])
orv/utils.py:242
↓ 2 callersFunctionrotm2euler
(R)
orv/dataset/dataset.py:90
↓ 2 callersFunctionsave_image
(image: torch.Tensor, path: pathlib.Path)
orv/dataset/encode_dataset.py:285
↓ 2 callersMethodsave_pretrained
( self, save_directory: Union[str, os.PathLike], is_main_process: bool = True,
orv/models/cogvideox_control.py:1056
↓ 2 callersFunctionzero_module
Zero out the parameters of a module and return it.
orv/models/components.py:112
↓ 1 callersFunctionCheckDeviceConsistency
orv/ops/include/pytorch_device_registry.hpp:74
↓ 1 callersMethodFind
orv/ops/include/pytorch_device_registry.hpp:39
↓ 1 callersFunctionGetFirstTensorDevice
orv/ops/include/pytorch_device_registry.hpp:62
↓ 1 callersMethod__getitem__
( self, index_or_res: int | tuple, camera_index: Optional[int] = 0, return_vid
orv/dataset/dataset.py:1014
↓ 1 callersMethod_aggregate_multivew
Args: data (list[dict]): len(data) = n_view.
orv/dataset/dataset.py:1779
↓ 1 callersMethod_apply_semantic_colormap
(self, semantic)
orv/dataset/dataset.py:642
↓ 1 callersFunction_calculate_frechet_distance
(mu1, sigma1, mu2, sigma2, eps=1e-6)
orv/pipeline/compute_metrics.py:171
↓ 1 callersFunction_compute_fid
(paths, device, batch_size, dims, num_frames)
orv/pipeline/compute_metrics.py:207
↓ 1 callersFunction_compute_fvd
(paths, device, batch_size, num_frames, num_videos: int = -1)
orv/pipeline/compute_metrics.py:306
↓ 1 callersFunction_compute_psnr_ssim
(pair, disable_tqdm = False)
orv/pipeline/compute_metrics.py:38
↓ 1 callersFunction_frechet_distance
(feats_fake: np.ndarray, feats_real: np.ndarray)
orv/pipeline/compute_metrics.py:278
↓ 1 callersMethod_get_positional_embeddings_v
( self, sample_views: int, device: Optional[torch.device] = None )
orv/models/cogvideox_control.py:659
↓ 1 callersMethod_get_sample_pos_embed_v
(self, n_view)
orv/models/cogvideox_control.py:679
↓ 1 callersFunction_get_t5_prompt_embeds
( tokenizer: T5Tokenizer, text_encoder: T5EncoderModel, prompt: Union[str, List[str]], num_vid
orv/dataset/encode_dataset.py:187
↓ 1 callersFunction_get_t5_prompt_embeds
( tokenizer: T5Tokenizer, text_encoder: T5EncoderModel, prompt: Union[str, List[str]], num_vid
orv/models/text_encoder.py:7
↓ 1 callersMethod_init_annos
(self)
orv/dataset/dataset.py:299
↓ 1 callersMethod_init_annos
(self)
orv/dataset/dataset.py:1563
↓ 1 callersFunction_pose_to_transform
(pose)
orv/dataset/prepare_dataset.py:911
↓ 1 callersFunction_preprocess_point_cloud
( pcd, max_nn=20, normals=True, )
orv/dataset/prepare_dataset.py:729
↓ 1 callersFunction_preprocess_points
(points_data)
orv/dataset/prepare_dataset.py:806
↓ 1 callersMethod_set_zeros
(self)
orv/models/cogvideox_control.py:625
↓ 1 callersFunctionalign_multiview_extrins
(data_dir, splits, rank: Optional[str] = None)
orv/dataset/prepare_dataset.py:1801
↓ 1 callersFunctionalpha2rotm
Alpha euler angle to rotation matrix.
orv/dataset/dataset.py:35
↓ 1 callersFunctionapply_semantic_colormap
(semantic)
orv/dataset/prepare_dataset.py:2044
↓ 1 callersMethodbackward
(ctx, grad_color, grad_language_feature, grad_radii, grad_depth, grad_alpha)
orv/ops/diff-gaussian-rasterization/diff_gaussian_rasterization/__init__.py:106
↓ 1 callersFunctionbeta2rotm
Beta euler angle to rotation matrix.
orv/dataset/dataset.py:45
↓ 1 callersMethodcompute_action_loss
( x: torch.Tensor, x_recon: torch.Tensor, loss_weight: dict, mask: Optional[to
orv/models/cogvideox_control.py:691
↓ 1 callersFunctioncompute_prompt_embeddings
( tokenizer: T5Tokenizer, text_encoder: T5EncoderModel, prompt: str, max_sequence_length: int,
orv/models/text_encoder.py:69
↓ 1 callersFunctioncompute_scale_and_shift_np
(prediction, target, mask)
orv/dataset/prepare_dataset.py:1871
↓ 1 callersFunctionconvert_scene_output_to_glb
(outdir, imgs, pts3d, mask, focals, cams2world, cam_size=0.05, show_cam=True,
orv/dataset/prepare_dataset.py:201
↓ 1 callersFunctioncreate_full_center_coords
(range, dim)
orv/dataset/gs_render.py:81
↓ 1 callersFunctiondownload
()
scripts/hf_model.py:56
↓ 1 callersMethodfetch_episode
(self, episode_id: str)
orv/dataset/dataset.py:1007
↓ 1 callersMethodfetch_index
(self, index: int)
orv/dataset/dataset.py:1004
↓ 1 callersFunctionfid_fvd
(gt_dir, pred_dir, fid: bool = True, fvd: bool = True)
orv/pipeline/compute_metrics.py:379
↓ 1 callersFunctionflatten_dict
(d, parent_key='', sep='.')
orv/utils.py:323
↓ 1 callersMethodforward
(self, h, action_hidden_states: Optional[torch.Tensor] = None)
orv/models/components.py:141
↓ 1 callersFunctiongamma2rotm
Gamma euler angle to rotation matrix.
orv/dataset/dataset.py:55
↓ 1 callersFunctiongenerate_colors
(n=60)
orv/dataset/dataset.py:144
↓ 1 callersFunctiongenerate_video
Generates a video based on the given prompt and saves it to the specified path. Parameters: - prompt (str): The description of the video
orv/pipeline/inference_control_to_video.py:19
↓ 1 callersFunctionget_3D_model_from_scene
extract 3D_model (glb file) from a reconstructed scene
orv/dataset/prepare_dataset.py:249
↓ 1 callersFunctionget_cameras
( data_dir: str, save_dir: str, splits: list, )
orv/dataset/prepare_dataset.py:598
↓ 1 callersFunctionget_captions
(data_dir: str, save_dir: str, splits: list, rank: Optional
orv/dataset/prepare_dataset.py:1490
↓ 1 callersFunctionget_labels
(data_dir: str, save_dir: str, splits: list, rank: Optional[str] = None)
orv/dataset/prepare_dataset.py:1044
↓ 1 callersMethodget_n_views_for_all_samples
(self)
orv/dataset/dataset.py:639
↓ 1 callersMethodget_n_views_for_all_samples
(self, train: bool = True)
orv/dataset/dataset.py:1762
↓ 1 callersFunctionget_optimizer
( params_to_optimize, optimizer_name: str = "adam", learning_rate: float = 1e-3, beta1: float
orv/utils.py:16
↓ 1 callersFunctionget_projection_matrix_c
(fx, fy, cx, cy, W, H, znear, zfar)
orv/dataset/gs_render.py:204
↓ 1 callersMethodget_ref_nums_for_all_samples
(self)
orv/dataset/dataset.py:636
↓ 1 callersFunctionget_render
(data_dir, splits: list, save_color: bool = True, rank: Optional[
orv/dataset/prepare_dataset.py:2024
↓ 1 callersFunctionget_resize_crop_region_for_grid
(src, tgt_width, tgt_height)
orv/utils.py:178
↓ 1 callersFunctionisRotm
(R)
orv/dataset/dataset.py:80
↓ 1 callersFunctionis_valid_image_imagelist
r""" Checks if the input is a valid image or list of images. The input can be one of the following formats: - A 4D tensor or numpy array
orv/models/components.py:204
↓ 1 callersFunctionload_i3d_pretrained
(device=torch.device('cpu'))
orv/pipeline/compute_metrics.py:297
↓ 1 callersFunctionlog_validation
( accelerator: Accelerator, pipe: Union[CogVideoXPipeline, CogVideoXImageToVideoPipelineTraj], con
orv/pipeline/train_cogvideox_control_to_video_sft.py:71
↓ 1 callersFunctionmain
()
orv/dataset/encode_dataset.py:417
↓ 1 callersFunctionmain
(config)
orv/pipeline/evaluation_control_to_video.py:115
↓ 1 callersFunctionmain
(config)
orv/pipeline/train_cogvideox_control_to_video_sft.py:161
↓ 1 callersFunctionnondeterministic_hard_voxelize_forward_impl
orv/ops/voxelize/voxelization.cpp:68
↓ 1 callersFunctionpair_videos
(gt_dir, pred_dir)
orv/pipeline/compute_metrics.py:83
↓ 1 callersFunctionparse_args
()
orv/pipeline/train_cogvideox_control_to_video_sft.py:1217
↓ 1 callersFunctionpnsr_ssim
(gt_dir, pred_dir, use_single=False, num_workers=8)
orv/pipeline/compute_metrics.py:107
↓ 1 callersFunctionpoints_to_voxels
(points: npt.NDArray | Tensor, voxel_size: list = [0.2, 0.2, 0.2], l
orv/dataset/prepare_dataset.py:137
next →1–100 of 239, ranked by callers