MCPcopy Create free account

hub / github.com/Yukun-Huang/DreamCube / functions

Functions331 in github.com/Yukun-Huang/DreamCube

Method__init__
(self, equirectangular)
utils/equi.py:47
Method__init__
( self, video_path: Union[str, Path], fps: int = 25, image_width: Optional[int
utils/video.py:24
Method__init__
( self, video_path: Union[str, Path], width: int = None, height: int = None,
utils/video.py:89
Method__init__
( self, video_path: Union[str, Path], fps: int = 25, )
utils/video.py:131
Method__init__
(self, dataset, rank=0, num_replicas=1, shuffle=True, seed=0, window_size=0.5)
utils/ops/misc.py:111
Method__init__
( self, height: int, width: int, cube_format: str, clip_output: bool =
external/equilib/cube2equi/base.py:47
Method__init__
( self, w_face: int, cube_format: str, z_down: bool = False, clip_outp
external/equilib/equi2cube/base.py:51
Method__init__
( self, height: int, width: int, fov_x: float, skew: float = 0.0,
external/equilib/equi2pers/base.py:38
Method__init__
( self, height: Optional[int] = None, width: Optional[int] = None, clip_output
external/equilib/equi2equi/base.py:34
Method__iter__
(self)
utils/ops/misc.py:124
Functionapply_custom_processors_for_transformer
( model, enable_sync_self_attn: bool = True, enable_sync_cross_attn: bool = False, enable_sync
models/multiplane_sync_legacy.py:476
Functionapply_custom_processors_for_transformer
( model, enable_sync_self_attn: bool = True, enable_sync_cross_attn: bool = False, enable_sync
models/multiplane_sync/processors_sana.py:84
Functionapply_custom_processors_for_unet
( model, enable_sync_self_attn: bool = False, enable_sync_cross_attn: bool = False, enable_syn
models/multiplane_sync/processors_videocrafter2.py:578
Functionapply_custom_processors_for_unet
( model, enable_sync_self_attn: bool = True, enable_sync_cross_attn: bool = False, enable_sync
models/multiplane_sync/processors_sd.py:40
Functionapply_custom_processors_for_vae
( model, mode: str = 'all', enable_sync_gn: bool = True, enable_sync_conv2d: bool = False,
models/multiplane_sync/processors_videocrafter2.py:612
Functionapply_custom_processors_for_vae
( model, mode: str = 'all', enable_sync_gn: bool = True, enable_sync_conv2d: bool = False,
models/multiplane_sync/processors_sd.py:73
Functionapply_custom_processors_for_vae
( model, mode: str = 'all', enable_sync_gn: bool = True, enable_sync_conv2d: bool = False,
models/multiplane_sync/processors_sana.py:44
Functionassert_shape
(tensor, ref_shape)
utils/ops/misc.py:81
Methodbackward
(ctx, grad_output)
utils/ops/grid_sample_gradfix.py:51
Methodbackward
(ctx, grad2_grad_input, grad2_grad_grid)
utils/ops/grid_sample_gradfix.py:67
Methodbackward
(ctx, grad_output)
utils/ops/grid_sample_gradfix.py:92
Methodbackward
(ctx, grad3_grad_input, grad3_grad_grid)
utils/ops/grid_sample_gradfix.py:108
Methodbackward
(ctx, dy)
utils/ops/resample_padding.py:186
Functionc2e
(cubemap, h, w, mode='bilinear', cube_format='dice')
external/py360convert/c2e.py:6
Functioncheck_ddp_consistency
(module, ignore_regex=None)
utils/ops/misc.py:179
Functioncheck_mesh_watertight
Load mesh from .obj file and check if it's watertight, also report boundary edges if it's not.
utils/mesh.py:4
Functionconcat_dice_mask
Args: dice: np.ndarray, (H, W, C), uint8 | Image.Image Returns: dice: np.ndarray, (H, W, C+1), uint8 | Image.Image
utils/cube.py:213
Functionconstant
(value, shape=None, dtype=None, device=None, memory_format=None)
utils/ops/misc.py:21
Functionconvert2horizon
Converts supported cubemap formats to horizon params: - cubemap - cube_format (str): ('horizon', 'dice', 'dict', 'list') return:
external/equilib/cube2equi/torch.py:68
Functionconvert2horizon
Converts supported cubemap formats to horizon params: - cubemap - cube_format (str): ('horizon', 'dice', 'dict', 'list') return:
external/equilib/cube2equi/numpy.py:57
Functionconvert_rgbd_cube_to_pcd
Converts panoramic RGBD data (image, distance, rays) into an Open3D mesh. Args: rgb: Input RGB image tensor (M, H, W, 3), uint8 or f
utils/pano_to_3d.py:945
Functionconvert_rgbd_equi_to_pcd
Converts panoramic RGBD data (image, distance, rays) into an Open3D mesh. Args: image: Input RGB image tensor (H, W, 3), uint8 or fl
utils/pano_to_3d.py:398
Functioncoor2uv
(coorxy, h, w)
external/py360convert/utils.py:117
Functioncopy_params_and_buffers
(src_module, dst_module, require_all=False)
utils/ops/misc.py:154
Functioncreate_rotation_matrix_at_once
Create rotation matrix at once" params: - roll, pitch, yaw (float): in radians - z_down (bool): flips pitch and yaw directions - dtyp
external/equilib/numpy_utils/rotation.py:91
Functioncreate_rotation_matrix_at_once
Create rotation matrix at once" params: - roll, pitch, yaw (float): in radians - z_down (bool): flips pitch and yaw directions - dtyp
external/equilib/torch_utils/rotation.py:81
Functioncreate_rotation_matrix_dep
Create rotation matrix NOTE: DEPRECATED
external/equilib/numpy_utils/rotation.py:160
Functioncreate_rotation_matrix_dep
Create Rotation Matrix NOTE: DEPRECATED
external/equilib/torch_utils/rotation.py:157
Functioncube_dice2h
(cube_dice)
external/py360convert/utils.py:215
Functioncube_dict2h
(cube_dict, face_k=['F', 'R', 'B', 'L', 'U', 'D'])
external/py360convert/utils.py:193
Functioncube_h2dice
(cube_h)
external/py360convert/utils.py:198
Functioncube_h2dict
(cube_h)
external/py360convert/utils.py:187
Functioncube_sync_ln_processor
(self)
models/multiplane_sync/sync_norm.py:125
Functioncube_sync_self_attn_processor
(self)
models/multiplane_sync_legacy.py:267
Functioncube_sync_self_attn_processor
(self)
models/multiplane_sync/sync_attn.py:6
Functioncube_video_sync_gn_processor
(self)
models/multiplane_sync/sync_norm.py:63
Functionddp_sync
(module, sync)
utils/ops/misc.py:168
Methoddecode_cube_latents
(self, latents: torch.Tensor)
models/dreamcube.py:78
Functiondecorator
(*args, **kwargs)
utils/ops/misc.py:100
Functiondump_vid
(video_path: Union[str, Path], frames: Iterable[Image.Image], fps: int = 25,)
utils/video.py:18
Functione2c
e_img: ndarray in shape of [H, W, *] face_w: int, the length of each face of the cubemap
external/py360convert/e2c.py:6
Functione2p
e_img: ndarray in shape of [H, W, *] fov_deg: scalar or (scalar, scalar) field of view in degree u_deg: horizon viewing angle in rang
external/py360convert/e2p.py:6
Functionequirect_facetype
0F 1R 2B 3L 4U 5D
external/py360convert/utils.py:47
Functionequirect_uvgrid
(h, w)
external/py360convert/utils.py:40
Functioneval_depth
(pred, target, eps=1e-9)
utils/depth.py:751
Methodexport_depth_to_16bit_png_one
(img, idx=None)
utils/depth.py:436
Methodflip
(self, flip=True)
utils/equi.py:71
Functionforward
Shape: - Input: :math:`(N, C, *)` where :math:`C=\text{num\_channels}` - Output: :math:`(N, C, *)` (same shape as input)
models/multiplane_sync_legacy.py:213
Functionforward
(input: torch.Tensor)
models/multiplane_sync/sync_conv2d.py:250
Functionforward
(input: torch.Tensor, *args, **kwargs)
models/multiplane_sync/utils.py:18
Functionforward
( hidden_states: torch.Tensor, encoder_hidden_states: Optional[torch.Tensor] = None, a
models/multiplane_sync/processors_videocrafter2.py:23
Functionforward
Shape: - Input: (B*M, C, H, W) or (B*M, C, HW) - Output: (B*M, C, H, W) or (B*M, C, HW)
models/multiplane_sync/sync_norm.py:8
Functionforward
(latent)
models/multiplane_sync/processors_sana.py:12
Functionforward
(hidden_states, encoder_hidden_states=None, attention_mask=None, **cross_attention_kwargs)
models/multiplane_sync/sync_attn.py:7
Methodforward
(ctx, input, grid)
utils/ops/grid_sample_gradfix.py:43
Methodforward
(ctx, grad_output, input, grid)
utils/ops/grid_sample_gradfix.py:60
Methodforward
(ctx, input, grid)
utils/ops/grid_sample_gradfix.py:84
Methodforward
(ctx, grad_output, input, grid)
utils/ops/grid_sample_gradfix.py:101
Methodforward
(ctx, x)
utils/ops/resample_padding.py:179
Functionfoward_wrapper
(orig_forward, note: str)
models/multiplane_sync/utils.py:16
Methodfrom_file
(cls, img_path)
utils/equi.py:51
Functionget_bounding_fov
( equi: torch.Tensor, rots: List[Dict[str, float]], height: int, width: int, fov_x: float,
external/equilib/equi2pers/torch.py:249
Functionget_bounding_fov
( equi: np.ndarray, rots: List[Dict[str, float]], height: int, width: int, fov_x: float,
external/equilib/equi2pers/numpy.py:231
Methodget_bounding_fov
(self, equi: ArrayLike, rots: Rot)
external/equilib/equi2pers/base.py:75
Functionget_patch_embed_forward
(self)
models/multiplane_sync/processors_sana.py:11
Functionget_plugin
(module_name, sources, headers=None, source_dir=None, **build_kwargs)
utils/ops/custom_ops.py:58
Functiongrid_sample
(input, grid)
utils/ops/grid_sample_gradfix.py:26
Functiongrid_sample
Torch grid sampling algorithm params: - img (torch.Tensor) - grid (torch.Tensor) - out (Optional[torch.Tensor]): defaults to None
external/equilib/grid_sample/torch/grid_sample.py:16
Functiongrid_sample
Numpy grid sampling algorithm params: - img (np.ndarray) - grid (np.ndarray) - out (np.ndarray) - mode (str): ('bilinear', 'bicub
external/equilib/grid_sample/numpy/grid_sample.py:12
Functionimages_to_equi_and_dice
Args: images: np.ndarray (B, 6, H, W, C) | torch.Tensor (B, 6, C, H, W) Returns: equis: np.ndarray, (B, H, W, C) dice
utils/cube.py:241
Functioninverse_sigmoid
(x)
utils/pano_to_3d.py:356
Methodmethod_custom
(image, cmap, bytes=False)
utils/depth.py:308
Methodmethod_matplotlib
(image, cmap, bytes=False)
utils/depth.py:287
Functionmkdirs
(path: Union[str, Path])
utils/video.py:11
Functionnative
Torch Grid Sample (default) - Uses `torch.nn.functional.grid_sample` - By far the best way to sample params: - img (torch.Tensor): T
external/equilib/grid_sample/torch/native.py:11
Functionpad_equi
(pano: torch.Tensor, padding: int)
utils/equi.py:12
Functionparams_and_buffers
(module)
utils/ops/misc.py:146
Functionpost_hook
(mod, _inputs, outputs)
utils/ops/misc.py:206
Functionpre_hook
(_mod, _inputs)
utils/ops/misc.py:204
Methodpreprocess
( self, image: PipelineImageInput, processing_resolution: Optional[int] = None,
utils/depth.py:197
Functionprint_module_summary
(module, inputs, max_nesting=3, skip_redundant=True)
utils/ops/misc.py:195
Functionprofiled_function
(fn)
utils/ops/misc.py:99
Methodpt_to_numpy
Convert a PyTorch tensor to a NumPy image.
utils/depth.py:50
Methodrelease
(self)
utils/video.py:124
Functionresample_padding
utils/ops/resample_padding.cpp:24
Methodrotate
(self, degree)
utils/equi.py:64
Functionrotate_cube
Args: cube: torch.Tensor, [b*m, c, h, w] or [b, m, c, h, w] yaw: int
utils/cube.py:65
Functionrotate_cube_by_shifts
Args: cube: torch.Tensor [b*m, c, h, w] or [b, m, c, h, w] | np.ndarray, [b*m, h, w, c] or [b, m, h, w, c] shifts: int
utils/cube.py:99
Functionrun
Run Cube2Equi params: - horizon (torch.Tensor) - height, widht (int): output equirectangular image's size - mode (str) - backend
external/equilib/cube2equi/torch.py:248
Functionrun
Run Cube2Equi params: - horizon (np.ndarray) - height, widht (int): output equirectangular image's size - mode (str) return:
external/equilib/cube2equi/numpy.py:254
← previousnext →201–300 of 331, ranked by callers