MCPcopy Create free account

hub / github.com/openai/point-e / functions

Functions254 in github.com/openai/point-e

↓ 1 callersFunctionopen_array
(path: str, arr_name: str)
point_e/evals/npz_stream.py:260
↓ 1 callersFunctionopen_npz_arrays
(path: str, arr_names: Sequence[str])
point_e/evals/npz_stream.py:186
↓ 1 callersMethodp_mean_variance
(self, model, *args, **kwargs)
point_e/diffusion/gaussian_diffusion.py:976
↓ 1 callersMethodp_sample
Sample x_{t-1} from the model at the given timestep. :param model: the model to sample from. :param x: the current tensor at
point_e/diffusion/gaussian_diffusion.py:407
↓ 1 callersFunctionplace_camera
(time, camera_pose_mode="random", camera_dist_min=2.0, camera_dist_max=2.0)
point_e/evals/scripts/blender_script.py:151
↓ 1 callersMethodpredict_sdf
Predict the SDF at the query points given the encoded point clouds. Each query point should be treated independently, only condition
point_e/models/sdf.py:61
↓ 1 callersFunctionpreprocess
(data, channel)
point_e/util/point_cloud.py:12
↓ 1 callersMethodq_mean_variance
Get the distribution q(x_t | x_0). :param x_start: the [N x C x ...] tensor of noiseless inputs. :param t: the number of dif
point_e/diffusion/gaussian_diffusion.py:201
↓ 1 callersFunctionrandomize_camera
(camera_dist=2.0)
point_e/evals/scripts/blender_script.py:134
↓ 1 callersMethodread_batch
(self, batch_size: int)
point_e/evals/npz_stream.py:138
↓ 1 callersMethodread_batch
(self, batch_size: int)
point_e/evals/npz_stream.py:217
↓ 1 callersFunctionrender_scene
(output_path, fast_mode: bool)
point_e/evals/scripts/blender_script.py:358
↓ 1 callersFunctionsample_and_group
Input: npoint: radius: nsample: xyz: input points position data, [B, N, 3] points: input points data, [B,
point_e/evals/pointnet2_utils.py:145
↓ 1 callersFunctionsample_and_group_all
Input: xyz: input points position data, [B, N, 3] points: input points data, [B, N, D] Return: new_xyz: sampled point
point_e/evals/pointnet2_utils.py:178
↓ 1 callersMethodsample_batch_progressive
( self, batch_size: int, model_kwargs: Dict[str, Any] )
point_e/diffusion/sampler.py:96
↓ 1 callersFunctionsave_rendering_dataset
( input_path: str, output_path: str, num_images: int, backend: str, light_mode: str, c
point_e/evals/scripts/blender_script.py:451
↓ 1 callersMethodscale_channels
(self, x: th.Tensor)
point_e/diffusion/gaussian_diffusion.py:922
↓ 1 callersFunctionscene_fov
()
point_e/evals/scripts/blender_script.py:418
↓ 1 callersFunctionscene_meshes
()
point_e/evals/scripts/blender_script.py:91
↓ 1 callersFunctionsetup_nodes
(output_path, capturing_material_alpha: bool = False)
point_e/evals/scripts/blender_script.py:292
↓ 1 callersMethodsigma_to_t
(self, sigma)
point_e/diffusion/k_diffusion.py:89
↓ 1 callersFunctionspace_timesteps
Create a list of timesteps to use from an original diffusion process, given the number of timesteps we want to take from equally-sized portio
point_e/diffusion/gaussian_diffusion.py:74
↓ 1 callersMethodsplit_model_output
( self, output: torch.Tensor, rescale_colors: bool = False, )
point_e/diffusion/sampler.py:207
↓ 1 callersMethodstream
(self, batch_size: int, keys: Sequence[str])
point_e/evals/npz_stream.py:86
↓ 1 callersMethodunscale_channels
(self, x: th.Tensor)
point_e/diffusion/gaussian_diffusion.py:933
↓ 1 callersFunctionwrite_camera_metadata
(path)
point_e/evals/scripts/blender_script.py:430
Method__call__
(self, x, ts, **kwargs)
point_e/diffusion/gaussian_diffusion.py:1000
Method__call__
( self, batch_size: int, images: Optional[Iterable[Optional[ImageType]]] = None,
point_e/models/pretrained_clip.py:237
Method__init__
(self, sigma_data: float = 0.5)
point_e/diffusion/k_diffusion.py:32
Method__init__
(self, model, diffusion)
point_e/diffusion/k_diffusion.py:80
Method__init__
( self, *, betas: Sequence[float], model_mean_type: str, model_var_typ
point_e/diffusion/gaussian_diffusion.py:144
Method__init__
(self, use_timesteps: Iterable[int], **kwargs)
point_e/diffusion/gaussian_diffusion.py:960
Method__init__
( self, device: torch.device, models: Sequence[nn.Module], diffusions: Sequenc
point_e/diffusion/sampler.py:26
Method__init__
(self, num_class, normal_channel=True, width_mult=1)
point_e/evals/pointnet2_cls_ssg.py:34
Method__init__
(self, glob_path: str)
point_e/evals/npz_stream.py:79
Method__init__
(self, arr_f, shape, dtype)
point_e/evals/npz_stream.py:143
Method__init__
(self, arr)
point_e/evals/npz_stream.py:166
Method__init__
(self, keys: List[str], readers: List[NpzArrayReader])
point_e/evals/npz_stream.py:213
Method__init__
( self, devices: List[Union[str, torch.device]], device_batch_size: int = 64,
point_e/evals/feature_extractor.py:52
Method__init__
(self, npoint, radius_list, nsample_list, in_channel, mlp_list)
point_e/evals/pointnet2_utils.py:245
Method__init__
(self, in_channel, mlp)
point_e/evals/pointnet2_utils.py:305
Method__init__
(self, mu: np.ndarray, sigma: np.ndarray)
point_e/evals/fid_is.py:17
Method__init__
( self, *, device: torch.device, dtype: torch.dtype, n_ctx: int = 4096
point_e/models/sdf.py:78
Method__init__
( self, *, device: torch.device, dtype: torch.dtype, n_ctx: int,
point_e/models/transformer.py:24
Method__init__
(self, *, device: torch.device, dtype: torch.dtype, heads: int, n_ctx: int)
point_e/models/transformer.py:66
Method__init__
( self, *, device: torch.device, dtype: torch.dtype, n_ctx: int,
point_e/models/transformer.py:88
Method__init__
( self, *, device: torch.device, dtype: torch.dtype, n_ctx: int,
point_e/models/transformer.py:119
Method__init__
( self, *, device: torch.device, dtype: torch.dtype, input_channels: i
point_e/models/transformer.py:156
Method__init__
( self, *, device: torch.device, dtype: torch.dtype, n_ctx: int = 1024
point_e/models/transformer.py:230
Method__init__
( self, *, device: torch.device, dtype: torch.dtype, n_ctx: int = 1024
point_e/models/transformer.py:291
Method__init__
( self, *, device: torch.device, dtype: torch.dtype, cond_input_channe
point_e/models/transformer.py:359
Method__init__
( self, *, device: torch.device, dtype: torch.dtype, n_ctx: int = 4096
point_e/models/transformer.py:413
Method__init__
( self, *, device: torch.device, dtype: torch.dtype, n_data: int,
point_e/models/perceiver.py:12
Method__init__
(self, *, device: torch.device, dtype: torch.dtype, heads: int, n_data: int)
point_e/models/perceiver.py:47
Method__init__
( self, *, device: torch.device, dtype: torch.dtype, n_data: int,
point_e/models/perceiver.py:71
Method__init__
( self, device: torch.device, dtype: Optional[torch.dtype] = torch.float32, en
point_e/models/pretrained_clip.py:19
Method__init__
(self, device: torch.device, **kwargs)
point_e/models/pretrained_clip.py:220
Methodbackward
(ctx, *output_grads)
point_e/models/checkpoint.py:43
Methodcached_model_kwargs
(self, batch_size: int, model_kwargs: Dict[str, Any])
point_e/models/transformer.py:317
Methodcached_model_kwargs
(self, batch_size: int, model_kwargs: Dict[str, Any])
point_e/models/transformer.py:440
Methodcalc_bpd_loop
Compute the entire variational lower-bound, measured in bits-per-dim, as well as other related quantities. :param model: the
point_e/diffusion/gaussian_diffusion.py:865
Methodcombine
(cls, *samplers: "PointCloudSampler")
point_e/diffusion/sampler.py:172
Methodcombine
(self, other: "PointCloud")
point_e/util/point_cloud.py:167
Methodcompute_dists
(idx: int)
point_e/util/point_cloud.py:109
Methodcompute_features
(i_batch)
point_e/evals/feature_extractor.py:102
Methodcondition_mean
(self, cond_fn, *args, **kwargs)
point_e/diffusion/gaussian_diffusion.py:982
Methodcondition_score
(self, cond_fn, *args, **kwargs)
point_e/diffusion/gaussian_diffusion.py:985
Functioncreate_default_materials
()
point_e/evals/scripts/blender_script.py:245
Methodddim_reverse_sample
Sample x_{t+1} from the model using DDIM reverse ODE.
point_e/diffusion/gaussian_diffusion.py:600
Methodddim_sample_loop
Generate samples from the model using DDIM. Same usage as p_sample_loop().
point_e/diffusion/gaussian_diffusion.py:638
Methoddefault_batch_size
Get a reasonable default number of query points for the model. In some cases, this might be the only supported size.
point_e/models/sdf.py:21
Methoddefault_batch_size
(self)
point_e/models/sdf.py:123
Methoddenoise
(self, x_t, sigmas, clip_denoised=True, model_kwargs=None)
point_e/diffusion/k_diffusion.py:98
Methoddevice
(self)
point_e/models/sdf.py:119
Functiondiffusion_from_config
(config: Dict[str, Any])
point_e/diffusion/configs.py:41
Methodelem_shape
(self)
point_e/evals/npz_stream.py:56
Methodembed_images
(self, xs: Iterable[Optional[ImageType]])
point_e/models/pretrained_clip.py:249
Methodembed_images_grid
(self, xs: Iterable[Optional[ImageType]])
point_e/models/pretrained_clip.py:257
Methodembed_text
(self, prompts: Iterable[str])
point_e/models/pretrained_clip.py:253
Methodencode_point_clouds
(self, point_clouds: torch.Tensor)
point_e/models/sdf.py:126
Methodfarthest_point_sample
Sample a subset of the point cloud that is evenly distributed in space. First, a random point is selected. Then each successive poin
point_e/util/point_cloud.py:82
Methodfeature_dim
(self)
point_e/evals/feature_extractor.py:29
Methodfeature_dim
(self)
point_e/evals/feature_extractor.py:77
Methodfeature_dim
(self)
point_e/models/pretrained_clip.py:47
Methodfeature_dim
(self)
point_e/models/pretrained_clip.py:226
Methodfeatures_and_preds
For a stream of point cloud batches, compute feature vectors and class predictions. :param point_clouds: a streamer for a sa
point_e/evals/feature_extractor.py:38
Functionfind_materials
()
point_e/evals/scripts/blender_script.py:254
Methodforward
(self, xyz, features=False)
point_e/evals/pointnet2_cls_ssg.py:71
Methodforward
(self, pred, target, trans_feat)
point_e/evals/pointnet2_cls_ssg.py:98
Methodforward
Input: xyz: input points position data, [B, C, N] points: input points data, [B, D, N] Return: ne
point_e/evals/pointnet2_utils.py:213
Methodforward
Input: xyz: input points position data, [B, C, N] points: input points data, [B, D, N] Return: ne
point_e/evals/pointnet2_utils.py:263
Methodforward
Input: xyz1: input points position data, [B, C, N] xyz2: sampled input points position data, [B, C, S] po
point_e/evals/pointnet2_utils.py:315
Methodforward
Predict the SDF at the coordinates x, given a batch of point clouds. Either point_clouds or encoded should be passed. Only exactly o
point_e/models/sdf.py:37
Methodforward
(ctx, run_function, length, *args)
point_e/models/checkpoint.py:34
Methodforward
(self, x)
point_e/models/transformer.py:44
Methodforward
(self, x)
point_e/models/transformer.py:61
Methodforward
(self, qkv)
point_e/models/transformer.py:73
Methodforward
(self, x: torch.Tensor)
point_e/models/transformer.py:112
Methodforward
(self, x: torch.Tensor)
point_e/models/transformer.py:149
Methodforward
:param x: an [N x C x T] tensor. :param t: an [N] tensor. :return: an [N x C' x T] tensor.
point_e/models/transformer.py:195
← previousnext →101–200 of 254, ranked by callers