MCPcopy Create free account

hub / github.com/GasaiYU/PartRM / types & classes

Types & classes71 in github.com/GasaiYU/PartRM

↓ 4 callersClassOptions
core/options.py:6
↓ 4 callersClassOptions
core/options_pm.py:7
↓ 4 callersClassResnetBlock
core/unet.py:53
↓ 3 callersClassMVAttention
core/unet.py:13
↓ 2 callersClassDinoWrapper
Dino v1 wrapper using huggingface transformer implementation.
preprocess/src/models/encoder/dino_wrapper.py:22
↓ 2 callersClassDownBlock
core/unet.py:107
↓ 2 callersClassFourierEmbedder
core/drag_embedding.py:6
↓ 2 callersClassLGM
core/models.py:15
↓ 2 callersClassMVDiffusion
preprocess/zero123plus/model.py:44
↓ 2 callersClassMidBlock
core/unet.py:152
↓ 2 callersClassRefOnlyNoisedUNet
preprocess/zero123plus/pipeline.py:81
↓ 2 callersClassTriplaneSynthesizer
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 callersClassTriplaneTransformer
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 callersClassUpBlock
core/unet.py:186
↓ 1 callersClassBasicTransformerBlock
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 callersClassDepthControlUNet
preprocess/zero123plus/pipeline.py:178
↓ 1 callersClassDragPositionNet
core/drag_embedding.py:23
↓ 1 callersClassDragPositionNetMultiScale
core/drag_embedding.py:76
↓ 1 callersClassDragPropagateDataset
preprocess/gen_propagate_drags.py:48
↓ 1 callersClassFlexiCubes
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 callersClassFlexiCubesGeometry
preprocess/src/models/geometry/rep_3d/flexicubes_geometry.py:35
↓ 1 callersClassGaussianRenderer
core/gs.py:16
↓ 1 callersClassImportanceRenderer
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 callersClassMemEffAttention
core/attention.py:67
↓ 1 callersClassMipRayMarcher2
preprocess/src/models/renderer/utils/ray_marcher.py:25
↓ 1 callersClassNeuralRender
preprocess/src/models/geometry/render/neural_render.py:69
↓ 1 callersClassOSGDecoder
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 callersClassOSGDecoder
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 callersClassPerspectiveCamera
preprocess/src/models/geometry/camera/perspective_camera.py:24
↓ 1 callersClassRaySampler
preprocess/src/models/renderer/utils/ray_sampler.py:22
↓ 1 callersClassReferenceOnlyAttnProc
preprocess/zero123plus/pipeline.py:44
↓ 1 callersClassSuperNet
preprocess/zero123plus/pipeline.py:229
↓ 1 callersClassUNet
core/unet.py:236
↓ 1 callersClassUNetWithMSDrag
core/unet.py:333
↓ 1 callersClassViTAttention
preprocess/src/models/encoder/dino.py:231
↓ 1 callersClassViTEmbeddings
Construct the CLS token, position and patch embeddings. Optionally, also the mask token.
preprocess/src/models/encoder/dino.py:34
↓ 1 callersClassViTEncoder
preprocess/src/models/encoder/dino.py:357
↓ 1 callersClassViTIntermediate
preprocess/src/models/encoder/dino.py:270
↓ 1 callersClassViTLayer
This corresponds to the Block class in the timm implementation.
preprocess/src/models/encoder/dino.py:305
↓ 1 callersClassViTOutput
preprocess/src/models/encoder/dino.py:286
↓ 1 callersClassViTPatchEmbeddings
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 callersClassViTPooler
preprocess/src/models/encoder/dino.py:538
↓ 1 callersClassViTSelfAttention
preprocess/src/models/encoder/dino.py:153
↓ 1 callersClassViTSelfOutput
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
ClassArgs
PartDrag4D/rendering/distributed.py:11
ClassAttention
core/attention.py:31
ClassCamera
preprocess/src/models/geometry/camera/__init__.py:13
ClassCrossAttention
core/attention.py:87
ClassDMTetGeometry
preprocess/src/models/geometry/rep_3d/dmtet.py:396
ClassDataModuleFromConfig
preprocess/src/data/objaverse_zero123plus.py:15
ClassDataModuleFromConfig
preprocess/src/data/objaverse.py:29
ClassGeometry
preprocess/src/models/geometry/rep_3d/__init__.py:13
ClassInstantMesh
Full model of the large reconstruction model.
preprocess/src/models/lrm_mesh.py:30
ClassInstantNeRF
Full model of the large reconstruction model.
preprocess/src/models/lrm.py:28
ClassMVRecon
preprocess/src/model_mesh.py:26
ClassMVRecon
preprocess/src/model.py:14
ClassMemEffCrossAttention
core/attention.py:137
ClassModuleListDict
preprocess/zero123plus/pipeline.py:219
ClassObjaverseData
preprocess/src/data/objaverse_zero123plus.py:60
ClassObjaverseData
preprocess/src/data/objaverse.py:74
ClassObjaverseHQEvalDataset
core/eval_dataset_objaverse_hq.py:30
ClassObjaverseHQTrainDataset
core/train_dataset_objaverse_hq.py:30
ClassOrthoRaySampler
preprocess/src/models/renderer/utils/ray_sampler.py:86
ClassPartDrag4DEvalDatset
core/eval_dataset_partdrag4d.py:28
ClassPartDrag4DTrainDatset
core/train_dataset_partdrag4d.py:28
ClassRenderer
preprocess/src/models/geometry/render/__init__.py:3
ClassTriplaneSynthesizer
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
ClassValidationData
preprocess/src/data/objaverse.py:241
ClassViTModel
preprocess/src/models/encoder/dino.py:449
ClassViTPreTrainedModel
An abstract class to handle weights initialization and a simple interface for downloading and loading pretrained models.
preprocess/src/models/encoder/dino.py:410
ClassZero123PlusPipeline
preprocess/zero123plus/pipeline.py:269