Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/openai/shap-e
/ types & classes
Types & classes
112 in github.com/openai/shap-e
⨍
Functions
571
◇
Types & classes
112
↓ 70 callers
Class
AttrDict
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 callers
Class
RayVolumeIntegral
shap_e/models/nerf/ray.py:208
↓ 7 callers
Class
DatasetIterator
shap_e/models/transmitter/channels_encoder.py:221
↓ 7 callers
Class
PointCloud
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 callers
Class
Transformer
shap_e/models/generation/transformer.py:114
↓ 6 callers
Class
Query
shap_e/models/query.py:8
↓ 5 callers
Class
MLP
shap_e/models/generation/transformer.py:47
↓ 5 callers
Class
ProjectiveCamera
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 callers
Class
StratifiedRaySampler
Instead of fixed intervals, a sample is drawn uniformly at random from each interval.
shap_e/models/nerf/ray.py:410
↓ 4 callers
Class
DifferentiableProjectiveCamera
Implements a batch, differentiable, standard pinhole camera
shap_e/models/nn/camera.py:37
↓ 4 callers
Class
SimplePerceiver
Only does cross attention
shap_e/models/generation/perceiver.py:116
↓ 3 callers
Class
PosEmbLinear
shap_e/models/nn/encoding.py:43
↓ 3 callers
Class
Rays
A ray in ray casting.
shap_e/rendering/raycast/types.py:13
↓ 3 callers
Class
TriMesh
shap_e/rendering/raycast/types.py:57
↓ 3 callers
Class
VolumeRange
shap_e/models/volume.py:12
↓ 2 callers
Class
BlenderViewData
Interact with a dataset zipfile exported by view_data.py.
shap_e/rendering/blender/view_data.py:12
↓ 2 callers
Class
GaussianDiffusion
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 callers
Class
MultiviewPointCloudEmbedding
shap_e/models/nn/encoding.py:116
↓ 2 callers
Class
RayCollisions
The result of casting N rays onto a mesh.
shap_e/rendering/raycast/types.py:26
↓ 2 callers
Class
RayVolumeIntegralResults
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 callers
Class
TorchMesh
A 3D triangle mesh with optional data at the vertices and faces.
shap_e/rendering/torch_mesh.py:10
↓ 1 callers
Class
BidirectionalLights
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 callers
Class
CLIPImageGridPointDiffusionTransformer
shap_e/models/generation/transformer.py:301
↓ 1 callers
Class
CLIPImageGridUpsamplePointDiffusionTransformer
shap_e/models/generation/transformer.py:425
↓ 1 callers
Class
CLIPImagePointDiffusionTransformer
shap_e/models/generation/transformer.py:239
↓ 1 callers
Class
ChannelsDecoder
shap_e/models/transmitter/base.py:176
↓ 1 callers
Class
ChannelsParamsProj
shap_e/models/transmitter/params_proj.py:138
↓ 1 callers
Class
ChannelsProj
shap_e/models/transmitter/params_proj.py:92
↓ 1 callers
Class
ClampDiffusionNoiseBottleneck
shap_e/models/transmitter/bottleneck.py:81
↓ 1 callers
Class
ClampNoiseBottleneck
shap_e/models/transmitter/bottleneck.py:68
↓ 1 callers
Class
DifferentiableCameraBatch
Annotate a differentiable camera with a multi-dimensional batch shape.
shap_e/models/nn/camera.py:130
↓ 1 callers
Class
GaussianToKarrasDenoiser
shap_e/diffusion/k_diffusion.py:79
↓ 1 callers
Class
IdentityLatentBottleneck
shap_e/models/transmitter/bottleneck.py:62
↓ 1 callers
Class
IdentityLatentWarp
shap_e/models/transmitter/bottleneck.py:37
↓ 1 callers
Class
ImageCLIP
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 callers
Class
ImportanceRaySampler
Given the initial estimate of densities, this samples more from regions/bins expected to have objects.
shap_e/models/nerf/ray.py:459
↓ 1 callers
Class
LinearParamsProj
shap_e/models/transmitter/params_proj.py:33
↓ 1 callers
Class
MLPParamsProj
shap_e/models/transmitter/params_proj.py:63
↓ 1 callers
Class
McLookupTable
shap_e/rendering/mc.py:206
↓ 1 callers
Class
MetaLinear
shap_e/models/nn/ops.py:114
↓ 1 callers
Class
MultiheadAttention
shap_e/models/generation/transformer.py:19
↓ 1 callers
Class
MultiheadCrossAttention
shap_e/models/generation/perceiver.py:13
↓ 1 callers
Class
MultiviewPoseEmbedding
shap_e/models/nn/encoding.py:60
↓ 1 callers
Class
MultiviewTransformerEncoder
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 callers
Class
NeRSTFRenderer
shap_e/models/nerstf/renderer.py:19
↓ 1 callers
Class
OneStepNeRFRenderer
Renders rays using stratified sampling only unlike vanilla NeRF. The same setup as NeRF++.
shap_e/models/nerf/renderer.py:199
↓ 1 callers
Class
PointCloudPerceiverChannelsEncoder
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 callers
Class
PointCloudPerceiverEncoder
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 callers
Class
PointCloudTransformerChannelsEncoder
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 callers
Class
PointCloudTransformerEncoder
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 callers
Class
PointDiffusionPerceiver
shap_e/models/generation/perceiver.py:161
↓ 1 callers
Class
PointDiffusionTransformer
shap_e/models/generation/transformer.py:151
↓ 1 callers
Class
PointSetEmbedding
shap_e/models/nn/ops.py:328
↓ 1 callers
Class
PooledMLP
shap_e/models/generation/pooled_mlp.py:7
↓ 1 callers
Class
QKVMultiheadAttention
shap_e/models/generation/transformer.py:61
↓ 1 callers
Class
QKVMultiheadCrossAttention
shap_e/models/generation/perceiver.py:50
↓ 1 callers
Class
ResBlock
shap_e/models/generation/pooled_mlp.py:41
↓ 1 callers
Class
ResidualAttentionBlock
shap_e/models/generation/transformer.py:83
↓ 1 callers
Class
ResidualCrossAttentionBlock
shap_e/models/generation/perceiver.py:77
↓ 1 callers
Class
STFRenderer
shap_e/models/stf/renderer.py:45
↓ 1 callers
Class
SirenSin
shap_e/models/nn/ops.py:36
↓ 1 callers
Class
SpacedDiffusion
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 callers
Class
SplitVectorDiffusion
shap_e/models/generation/latent_diffusion.py:7
↓ 1 callers
Class
Tan2LatentWarp
shap_e/models/transmitter/bottleneck.py:47
↓ 1 callers
Class
Transmitter
shap_e/models/transmitter/base.py:131
↓ 1 callers
Class
TriMesh
A 3D triangle mesh with optional data at the vertices and faces.
shap_e/rendering/mesh.py:11
↓ 1 callers
Class
TwoStepNeRFRenderer
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 callers
Class
UpsamplePointDiffusionTransformer
shap_e/models/generation/transformer.py:371
↓ 1 callers
Class
VectorDecoder
shap_e/models/transmitter/base.py:145
↓ 1 callers
Class
_WrappedModel
shap_e/diffusion/gaussian_diffusion.py:1046
Class
BoundingBoxVolume
Axis-aligned bounding box defined by the two opposite corners.
shap_e/models/volume.py:89
Class
Camera
An object describing how a camera corresponds to pixels in an image.
shap_e/rendering/view_data.py:9
Class
ChannelsEncoder
shap_e/models/transmitter/base.py:82
Class
CheckpointFunction
shap_e/models/nn/checkpoint.py:29
Class
CheckpointFunctionGradFunction
shap_e/models/nn/checkpoint.py:59
Class
DifferentiableCamera
An object describing how a camera corresponds to pixels in an image.
shap_e/models/nn/camera.py:12
Class
Encoder
shap_e/models/transmitter/base.py:14
Class
FrozenImageCLIP
shap_e/models/generation/pretrained_clip.py:219
Class
KarrasDenoiser
shap_e/diffusion/k_diffusion.py:31
Class
LatentBottleneck
shap_e/models/transmitter/bottleneck.py:12
Class
LatentWarp
shap_e/models/transmitter/bottleneck.py:23
Class
LayerNorm
shap_e/models/nn/ops.py:311
Class
MLP
shap_e/models/nn/ops.py:223
Class
MLPDensitySDFModel
shap_e/models/nerstf/mlp.py:11
Class
MLPModel
shap_e/models/stf/mlp.py:17
Class
MLPNeRFModel
shap_e/models/nerf/model.py:83
Class
MLPNeRSTFModel
shap_e/models/nerstf/mlp.py:45
Class
MLPSDFModel
shap_e/models/stf/mlp.py:183
Class
MLPTextureFieldModel
shap_e/models/stf/mlp.py:198
Class
MemoryViewData
A ViewData that is implemented in memory.
shap_e/rendering/view_data.py:186
Class
MetaMLP
shap_e/models/nn/ops.py:260
Class
MetaModule
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
Class
Model
shap_e/models/stf/base.py:11
Class
NeRFModel
Parametric scene representation whose outputs are integrated by NeRFRenderer
shap_e/models/nerf/model.py:18
Class
ParamsProj
shap_e/models/transmitter/params_proj.py:21
Class
PerceiverChannelsEncoder
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
Class
PerceiverEncoder
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
Class
PointNetFeaturePropagation
shap_e/models/nn/pointnet2_utils.py:318
Class
PointNetSetAbstraction
shap_e/models/nn/pointnet2_utils.py:212
Class
PointNetSetAbstractionMsg
shap_e/models/nn/pointnet2_utils.py:258
next →
1–100 of 112, ranked by callers