MCPcopy Create free account

hub / github.com/openai/shap-e / types & classes

Types & classes112 in github.com/openai/shap-e

↓ 70 callersClassAttrDict
An attribute dictionary that automatically handles nested keys joined by "/". Originally copied from: https://stackoverflow.com/questions/30
shap_e/util/collections.py:8
↓ 8 callersClassRayVolumeIntegral
shap_e/models/nerf/ray.py:208
↓ 7 callersClassDatasetIterator
shap_e/models/transmitter/channels_encoder.py:221
↓ 7 callersClassPointCloud
An array of points sampled on a surface. Each point may have zero or more channel attributes. :param coords: an [N x 3] array of point c
shap_e/rendering/point_cloud.py:23
↓ 7 callersClassTransformer
shap_e/models/generation/transformer.py:114
↓ 6 callersClassQuery
shap_e/models/query.py:8
↓ 5 callersClassMLP
shap_e/models/generation/transformer.py:47
↓ 5 callersClassProjectiveCamera
A Camera implementation for a standard pinhole camera. The camera rays shoot away from the origin in the z direction, with the x and y d
shap_e/rendering/view_data.py:68
↓ 5 callersClassStratifiedRaySampler
Instead of fixed intervals, a sample is drawn uniformly at random from each interval.
shap_e/models/nerf/ray.py:410
↓ 4 callersClassDifferentiableProjectiveCamera
Implements a batch, differentiable, standard pinhole camera
shap_e/models/nn/camera.py:37
↓ 4 callersClassSimplePerceiver
Only does cross attention
shap_e/models/generation/perceiver.py:116
↓ 3 callersClassPosEmbLinear
shap_e/models/nn/encoding.py:43
↓ 3 callersClassRays
A ray in ray casting.
shap_e/rendering/raycast/types.py:13
↓ 3 callersClassTriMesh
shap_e/rendering/raycast/types.py:57
↓ 3 callersClassVolumeRange
shap_e/models/volume.py:12
↓ 2 callersClassBlenderViewData
Interact with a dataset zipfile exported by view_data.py.
shap_e/rendering/blender/view_data.py:12
↓ 2 callersClassGaussianDiffusion
Utilities for training and sampling diffusion models. Ported directly from here: https://github.com/hojonathanho/diffusion/blob/1e0dceb3
shap_e/diffusion/gaussian_diffusion.py:175
↓ 2 callersClassMultiviewPointCloudEmbedding
shap_e/models/nn/encoding.py:116
↓ 2 callersClassRayCollisions
The result of casting N rays onto a mesh.
shap_e/rendering/raycast/types.py:26
↓ 2 callersClassRayVolumeIntegralResults
Stores the relevant state and results of integrate( lambda t: density(t) * channels(t) * transmittance(t), [t0,
shap_e/models/nerf/ray.py:133
↓ 2 callersClassTorchMesh
A 3D triangle mesh with optional data at the vertices and faces.
shap_e/rendering/torch_mesh.py:10
↓ 1 callersClassBidirectionalLights
Adapted from here, but effectively shines the light in both positive and negative directions: https://github.com/facebookresearch/pytorch3d/b
shap_e/rendering/pytorch3d_util.py:233
↓ 1 callersClassCLIPImageGridPointDiffusionTransformer
shap_e/models/generation/transformer.py:301
↓ 1 callersClassCLIPImageGridUpsamplePointDiffusionTransformer
shap_e/models/generation/transformer.py:425
↓ 1 callersClassCLIPImagePointDiffusionTransformer
shap_e/models/generation/transformer.py:239
↓ 1 callersClassChannelsDecoder
shap_e/models/transmitter/base.py:176
↓ 1 callersClassChannelsParamsProj
shap_e/models/transmitter/params_proj.py:138
↓ 1 callersClassChannelsProj
shap_e/models/transmitter/params_proj.py:92
↓ 1 callersClassClampDiffusionNoiseBottleneck
shap_e/models/transmitter/bottleneck.py:81
↓ 1 callersClassClampNoiseBottleneck
shap_e/models/transmitter/bottleneck.py:68
↓ 1 callersClassDifferentiableCameraBatch
Annotate a differentiable camera with a multi-dimensional batch shape.
shap_e/models/nn/camera.py:130
↓ 1 callersClassGaussianToKarrasDenoiser
shap_e/diffusion/k_diffusion.py:79
↓ 1 callersClassIdentityLatentBottleneck
shap_e/models/transmitter/bottleneck.py:62
↓ 1 callersClassIdentityLatentWarp
shap_e/models/transmitter/bottleneck.py:37
↓ 1 callersClassImageCLIP
A wrapper around a pre-trained CLIP model that automatically handles batches of texts, images, and embeddings.
shap_e/models/generation/pretrained_clip.py:13
↓ 1 callersClassImportanceRaySampler
Given the initial estimate of densities, this samples more from regions/bins expected to have objects.
shap_e/models/nerf/ray.py:459
↓ 1 callersClassLinearParamsProj
shap_e/models/transmitter/params_proj.py:33
↓ 1 callersClassMLPParamsProj
shap_e/models/transmitter/params_proj.py:63
↓ 1 callersClassMcLookupTable
shap_e/rendering/mc.py:206
↓ 1 callersClassMetaLinear
shap_e/models/nn/ops.py:114
↓ 1 callersClassMultiheadAttention
shap_e/models/generation/transformer.py:19
↓ 1 callersClassMultiheadCrossAttention
shap_e/models/generation/perceiver.py:13
↓ 1 callersClassMultiviewPoseEmbedding
shap_e/models/nn/encoding.py:60
↓ 1 callersClassMultiviewTransformerEncoder
Encode cameras and views using a transformer model with extra output token(s) used to extract a latent vector.
shap_e/models/transmitter/multiview_encoder.py:16
↓ 1 callersClassNeRSTFRenderer
shap_e/models/nerstf/renderer.py:19
↓ 1 callersClassOneStepNeRFRenderer
Renders rays using stratified sampling only unlike vanilla NeRF. The same setup as NeRF++.
shap_e/models/nerf/renderer.py:199
↓ 1 callersClassPointCloudPerceiverChannelsEncoder
Encode point clouds using a transformer model with an extra output token used to extract a latent vector.
shap_e/models/transmitter/channels_encoder.py:286
↓ 1 callersClassPointCloudPerceiverEncoder
Encode point clouds using a transformer model with an extra output token used to extract a latent vector.
shap_e/models/transmitter/pc_encoder.py:196
↓ 1 callersClassPointCloudTransformerChannelsEncoder
Encode point clouds using a transformer model with an extra output token used to extract a latent vector.
shap_e/models/transmitter/channels_encoder.py:261
↓ 1 callersClassPointCloudTransformerEncoder
Encode point clouds using a transformer model with an extra output token used to extract a latent vector.
shap_e/models/transmitter/pc_encoder.py:21
↓ 1 callersClassPointDiffusionPerceiver
shap_e/models/generation/perceiver.py:161
↓ 1 callersClassPointDiffusionTransformer
shap_e/models/generation/transformer.py:151
↓ 1 callersClassPointSetEmbedding
shap_e/models/nn/ops.py:328
↓ 1 callersClassPooledMLP
shap_e/models/generation/pooled_mlp.py:7
↓ 1 callersClassQKVMultiheadAttention
shap_e/models/generation/transformer.py:61
↓ 1 callersClassQKVMultiheadCrossAttention
shap_e/models/generation/perceiver.py:50
↓ 1 callersClassResBlock
shap_e/models/generation/pooled_mlp.py:41
↓ 1 callersClassResidualAttentionBlock
shap_e/models/generation/transformer.py:83
↓ 1 callersClassResidualCrossAttentionBlock
shap_e/models/generation/perceiver.py:77
↓ 1 callersClassSTFRenderer
shap_e/models/stf/renderer.py:45
↓ 1 callersClassSirenSin
shap_e/models/nn/ops.py:36
↓ 1 callersClassSpacedDiffusion
A diffusion process which can skip steps in a base diffusion process. :param use_timesteps: (unordered) timesteps from the original diffusion
shap_e/diffusion/gaussian_diffusion.py:1004
↓ 1 callersClassSplitVectorDiffusion
shap_e/models/generation/latent_diffusion.py:7
↓ 1 callersClassTan2LatentWarp
shap_e/models/transmitter/bottleneck.py:47
↓ 1 callersClassTransmitter
shap_e/models/transmitter/base.py:131
↓ 1 callersClassTriMesh
A 3D triangle mesh with optional data at the vertices and faces.
shap_e/rendering/mesh.py:11
↓ 1 callersClassTwoStepNeRFRenderer
Coarse and fine-grained rendering as proposed by NeRF. This class additionally supports background rendering like NeRF++.
shap_e/models/nerf/renderer.py:15
↓ 1 callersClassUpsamplePointDiffusionTransformer
shap_e/models/generation/transformer.py:371
↓ 1 callersClassVectorDecoder
shap_e/models/transmitter/base.py:145
↓ 1 callersClass_WrappedModel
shap_e/diffusion/gaussian_diffusion.py:1046
ClassBoundingBoxVolume
Axis-aligned bounding box defined by the two opposite corners.
shap_e/models/volume.py:89
ClassCamera
An object describing how a camera corresponds to pixels in an image.
shap_e/rendering/view_data.py:9
ClassChannelsEncoder
shap_e/models/transmitter/base.py:82
ClassCheckpointFunction
shap_e/models/nn/checkpoint.py:29
ClassCheckpointFunctionGradFunction
shap_e/models/nn/checkpoint.py:59
ClassDifferentiableCamera
An object describing how a camera corresponds to pixels in an image.
shap_e/models/nn/camera.py:12
ClassEncoder
shap_e/models/transmitter/base.py:14
ClassFrozenImageCLIP
shap_e/models/generation/pretrained_clip.py:219
ClassKarrasDenoiser
shap_e/diffusion/k_diffusion.py:31
ClassLatentBottleneck
shap_e/models/transmitter/bottleneck.py:12
ClassLatentWarp
shap_e/models/transmitter/bottleneck.py:23
ClassLayerNorm
shap_e/models/nn/ops.py:311
ClassMLP
shap_e/models/nn/ops.py:223
ClassMLPDensitySDFModel
shap_e/models/nerstf/mlp.py:11
ClassMLPModel
shap_e/models/stf/mlp.py:17
ClassMLPNeRFModel
shap_e/models/nerf/model.py:83
ClassMLPNeRSTFModel
shap_e/models/nerstf/mlp.py:45
ClassMLPSDFModel
shap_e/models/stf/mlp.py:183
ClassMLPTextureFieldModel
shap_e/models/stf/mlp.py:198
ClassMemoryViewData
A ViewData that is implemented in memory.
shap_e/rendering/view_data.py:186
ClassMetaMLP
shap_e/models/nn/ops.py:260
ClassMetaModule
Base class for PyTorch meta-learning modules. These modules accept an additional argument `params` in their `forward` method. Notes
shap_e/models/nn/meta.py:82
ClassModel
shap_e/models/stf/base.py:11
ClassNeRFModel
Parametric scene representation whose outputs are integrated by NeRFRenderer
shap_e/models/nerf/model.py:18
ClassParamsProj
shap_e/models/transmitter/params_proj.py:21
ClassPerceiverChannelsEncoder
Encode point clouds using a perceiver model with an extra output token used to extract a latent vector.
shap_e/models/transmitter/channels_encoder.py:101
ClassPerceiverEncoder
Encode point clouds using a perceiver model with an extra output token used to extract a latent vector.
shap_e/models/transmitter/pc_encoder.py:90
ClassPointNetFeaturePropagation
shap_e/models/nn/pointnet2_utils.py:318
ClassPointNetSetAbstraction
shap_e/models/nn/pointnet2_utils.py:212
ClassPointNetSetAbstractionMsg
shap_e/models/nn/pointnet2_utils.py:258
next →1–100 of 112, ranked by callers