Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/GasaiYU/PartRM
/ types & classes
Types & classes
71 in github.com/GasaiYU/PartRM
⨍
Functions
389
◇
Types & classes
71
↓ 4 callers
Class
Options
core/options.py:6
↓ 4 callers
Class
Options
core/options_pm.py:7
↓ 4 callers
Class
ResnetBlock
core/unet.py:53
↓ 3 callers
Class
MVAttention
core/unet.py:13
↓ 2 callers
Class
DinoWrapper
Dino v1 wrapper using huggingface transformer implementation.
preprocess/src/models/encoder/dino_wrapper.py:22
↓ 2 callers
Class
DownBlock
core/unet.py:107
↓ 2 callers
Class
FourierEmbedder
core/drag_embedding.py:6
↓ 2 callers
Class
LGM
core/models.py:15
↓ 2 callers
Class
MVDiffusion
preprocess/zero123plus/model.py:44
↓ 2 callers
Class
MidBlock
core/unet.py:152
↓ 2 callers
Class
RefOnlyNoisedUNet
preprocess/zero123plus/pipeline.py:81
↓ 2 callers
Class
TriplaneSynthesizer
Synthesizer that renders a triplane volume with planes and a camera. Reference: EG3D: https://github.com/NVlabs/eg3d/blob/main/eg3d/
preprocess/src/models/renderer/synthesizer.py:61
↓ 2 callers
Class
TriplaneTransformer
Transformer with condition that generates a triplane representation. Reference: Timm: https://github.com/huggingface/pytorch-image-m
preprocess/src/models/decoder/transformer.py:66
↓ 2 callers
Class
UpBlock
core/unet.py:186
↓ 1 callers
Class
BasicTransformerBlock
Transformer block that takes in a cross-attention condition and another modulation vector applied to sub-blocks.
preprocess/src/models/decoder/transformer.py:20
↓ 1 callers
Class
DepthControlUNet
preprocess/zero123plus/pipeline.py:178
↓ 1 callers
Class
DragPositionNet
core/drag_embedding.py:23
↓ 1 callers
Class
DragPositionNetMultiScale
core/drag_embedding.py:76
↓ 1 callers
Class
DragPropagateDataset
preprocess/gen_propagate_drags.py:48
↓ 1 callers
Class
FlexiCubes
This class implements the FlexiCubes method for extracting meshes from scalar fields. It maintains a series of lookup tables and indices to
preprocess/src/models/geometry/rep_3d/flexicubes.py:16
↓ 1 callers
Class
FlexiCubesGeometry
preprocess/src/models/geometry/rep_3d/flexicubes_geometry.py:35
↓ 1 callers
Class
GaussianRenderer
core/gs.py:16
↓ 1 callers
Class
ImportanceRenderer
Modified original version to filter out-of-box samples as TensoRF does. Reference: TensoRF: https://github.com/apchenstu/TensoRF/blo
preprocess/src/models/renderer/utils/renderer.py:100
↓ 1 callers
Class
MemEffAttention
core/attention.py:67
↓ 1 callers
Class
MipRayMarcher2
preprocess/src/models/renderer/utils/ray_marcher.py:25
↓ 1 callers
Class
NeuralRender
preprocess/src/models/geometry/render/neural_render.py:69
↓ 1 callers
Class
OSGDecoder
Triplane decoder that gives RGB and sigma values from sampled features. Using ReLU here instead of Softplus in the original implementation.
preprocess/src/models/renderer/synthesizer_mesh.py:15
↓ 1 callers
Class
OSGDecoder
Triplane decoder that gives RGB and sigma values from sampled features. Using ReLU here instead of Softplus in the original implementation.
preprocess/src/models/renderer/synthesizer.py:17
↓ 1 callers
Class
PerspectiveCamera
preprocess/src/models/geometry/camera/perspective_camera.py:24
↓ 1 callers
Class
RaySampler
preprocess/src/models/renderer/utils/ray_sampler.py:22
↓ 1 callers
Class
ReferenceOnlyAttnProc
preprocess/zero123plus/pipeline.py:44
↓ 1 callers
Class
SuperNet
preprocess/zero123plus/pipeline.py:229
↓ 1 callers
Class
UNet
core/unet.py:236
↓ 1 callers
Class
UNetWithMSDrag
core/unet.py:333
↓ 1 callers
Class
ViTAttention
preprocess/src/models/encoder/dino.py:231
↓ 1 callers
Class
ViTEmbeddings
Construct the CLS token, position and patch embeddings. Optionally, also the mask token.
preprocess/src/models/encoder/dino.py:34
↓ 1 callers
Class
ViTEncoder
preprocess/src/models/encoder/dino.py:357
↓ 1 callers
Class
ViTIntermediate
preprocess/src/models/encoder/dino.py:270
↓ 1 callers
Class
ViTLayer
This corresponds to the Block class in the timm implementation.
preprocess/src/models/encoder/dino.py:305
↓ 1 callers
Class
ViTOutput
preprocess/src/models/encoder/dino.py:286
↓ 1 callers
Class
ViTPatchEmbeddings
This class turns `pixel_values` of shape `(batch_size, num_channels, height, width)` into the initial `hidden_states` (patch embeddings) of s
preprocess/src/models/encoder/dino.py:114
↓ 1 callers
Class
ViTPooler
preprocess/src/models/encoder/dino.py:538
↓ 1 callers
Class
ViTSelfAttention
preprocess/src/models/encoder/dino.py:153
↓ 1 callers
Class
ViTSelfOutput
The residual connection is defined in ViTLayer instead of here (as is the case with other models), due to the layernorm applied before each b
preprocess/src/models/encoder/dino.py:213
Class
Args
PartDrag4D/rendering/distributed.py:11
Class
Attention
core/attention.py:31
Class
Camera
preprocess/src/models/geometry/camera/__init__.py:13
Class
CrossAttention
core/attention.py:87
Class
DMTetGeometry
preprocess/src/models/geometry/rep_3d/dmtet.py:396
Class
DataModuleFromConfig
preprocess/src/data/objaverse_zero123plus.py:15
Class
DataModuleFromConfig
preprocess/src/data/objaverse.py:29
Class
Geometry
preprocess/src/models/geometry/rep_3d/__init__.py:13
Class
InstantMesh
Full model of the large reconstruction model.
preprocess/src/models/lrm_mesh.py:30
Class
InstantNeRF
Full model of the large reconstruction model.
preprocess/src/models/lrm.py:28
Class
MVRecon
preprocess/src/model_mesh.py:26
Class
MVRecon
preprocess/src/model.py:14
Class
MemEffCrossAttention
core/attention.py:137
Class
ModuleListDict
preprocess/zero123plus/pipeline.py:219
Class
ObjaverseData
preprocess/src/data/objaverse_zero123plus.py:60
Class
ObjaverseData
preprocess/src/data/objaverse.py:74
Class
ObjaverseHQEvalDataset
core/eval_dataset_objaverse_hq.py:30
Class
ObjaverseHQTrainDataset
core/train_dataset_objaverse_hq.py:30
Class
OrthoRaySampler
preprocess/src/models/renderer/utils/ray_sampler.py:86
Class
PartDrag4DEvalDatset
core/eval_dataset_partdrag4d.py:28
Class
PartDrag4DTrainDatset
core/train_dataset_partdrag4d.py:28
Class
Renderer
preprocess/src/models/geometry/render/__init__.py:3
Class
TriplaneSynthesizer
Synthesizer that renders a triplane volume with planes and a camera. Reference: EG3D: https://github.com/NVlabs/eg3d/blob/main/eg3d/
preprocess/src/models/renderer/synthesizer_mesh.py:93
Class
ValidationData
preprocess/src/data/objaverse.py:241
Class
ViTModel
preprocess/src/models/encoder/dino.py:449
Class
ViTPreTrainedModel
An abstract class to handle weights initialization and a simple interface for downloading and loading pretrained models.
preprocess/src/models/encoder/dino.py:410
Class
Zero123PlusPipeline
preprocess/zero123plus/pipeline.py:269