MCPcopy Create free account

hub / github.com/Zhiyuan-R/Tiger-Diffusion / types & classes

Types & classes46 in github.com/Zhiyuan-R/Tiger-Diffusion

↓ 7 callersClassSharedMLP
modules/shared_mlp.py:11
↓ 4 callersClassShapeNet15kPointClouds
datasets/shapenet_data_pc.py:201
↓ 3 callersClassAttention
modules/pvconv.py:16
↓ 3 callersClassSwish
modules/se.py:5
↓ 3 callersClassSwish
modules/pvconv.py:11
↓ 2 callersClassPointNetFPModule
modules/pointnet.py:96
↓ 2 callersClassSE3d
modules/se.py:8
↓ 2 callersClassVoxelization
modules/voxelization.py:9
↓ 1 callersClassAttention
model/transformer_branch.py:74
↓ 1 callersClassBallQuery
modules/ball_query.py:9
↓ 1 callersClassDepthToSingleViewPoints
render a view then save mask
datasets/shapenet_data_sv.py:225
↓ 1 callersClassDiT
Diffusion model with a Transformer backbone.
model/transformer_branch.py:166
↓ 1 callersClassDiTBlock
A DiT block with adaptive layer norm zero (adaLN-Zero) conditioning.
model/transformer_branch.py:122
↓ 1 callersClassFinalLayer
The final layer of DiT.
model/transformer_branch.py:147
↓ 1 callersClassGaussianDiffusion
test_generation.py:53
↓ 1 callersClassGaussianDiffusion
train_generation.py:99
↓ 1 callersClassModel
test_generation.py:273
↓ 1 callersClassModel
train_generation.py:402
↓ 1 callersClassPVCNN2
test_generation.py:249
↓ 1 callersClassPointCloudMasks
render a view then save mask
datasets/shapenet_data_pc.py:237
↓ 1 callersClassSwish
modules/shared_mlp.py:7
↓ 1 callersClassTiger_Transformer_custom
train_generation.py:378
↓ 1 callersClassTimestepEmbedder
Embeds scalar timesteps into vector representations.
model/transformer_branch.py:19
↓ 1 callersClasschamfer_3DDist
metrics/ChamferDistancePytorch/chamfer3D/dist_chamfer_3D.py:69
ClassAvgVoxelization
modules/functional/voxelization.py:8
ClassEarthMoverDistanceFunction
metrics/PyTorchEMD/emd.py:5
ClassFrustumPointNetLoss
modules/frustum.py:11
ClassGANdatasetPartNet
datasets/partnet.py:106
ClassGather
modules/functional/sampling.py:10
ClassGrouping
modules/functional/grouping.py:8
ClassKLLoss
modules/loss.py:8
ClassNeighborInterpolation
modules/functional/interpolatation.py:8
ClassPVCNN2Base
model/pvcnn_generation.py:172
ClassPVConv
modules/pvconv.py:64
ClassPVConvReLU
modules/pvconv.py:100
ClassPointNetAModule
modules/pointnet.py:11
ClassPointNetSAModule
modules/pointnet.py:49
ClassShapeNet_Multiview_Points
datasets/shapenet_data_sv.py:51
ClassTiger_Transformer
model/tiger.py:175
ClassTrilinearDevoxelization
modules/functional/devoxelization.py:8
ClassUniform15KPC
datasets/shapenet_data_pc.py:38
Classchamfer_2DDist
metrics/ChamferDistancePytorch/chamfer2D/dist_chamfer_2D.py:66
Classchamfer_2DFunction
metrics/ChamferDistancePytorch/chamfer2D/dist_chamfer_2D.py:25
Classchamfer_3DFunction
metrics/ChamferDistancePytorch/chamfer3D/dist_chamfer_3D.py:28
Classchamfer_5DDist
metrics/ChamferDistancePytorch/chamfer5D/dist_chamfer_5D.py:68
Classchamfer_5DFunction
metrics/ChamferDistancePytorch/chamfer5D/dist_chamfer_5D.py:27