MCPcopy Create free account

hub / github.com/AIR-DISCOVER/Omni-PQ / types & classes

Types & classes63 in github.com/AIR-DISCOVER/Omni-PQ

↓ 7 callersClassScannetDatasetConfig
scannet/model_util_scannet.py:14
↓ 6 callersClassScannetDetectionDataset
scannet/scannet_detection_dataset.py:36
↓ 5 callersClassPointnetSAModuleVotes
Modified based on _PointnetSAModuleBase and PointnetSAModuleMSG with extra support for returning point indices for getting their GT votes
pointnet2/pointnet2_modules.py:164
↓ 4 callersClassARKitSceneDataset
ARKitScenes/arkitscenes_dataset.py:42
↓ 3 callersClassMultiheadAttention
r"""Allows the model to jointly attend to information from different representation subspaces. See reference: Attention Is All You Need .
models/utils/multi_head_attention.py:12
↓ 3 callersClassQUADAPCalculator
Calculating Average Precision
models/ap_helper_pq.py:579
↓ 2 callersClassAPCalculator
Calculating Average Precision
models/ap_helper_pq.py:520
↓ 2 callersClassBatchNorm1d
pointnet2/pytorch_utils.py:49
↓ 2 callersClassDoubleLoaderTest
double_loader.py:32
↓ 2 callersClassFitRunner
fit.py:70
↓ 2 callersClassPQ_Transformer
models/pq_transformer.py:123
↓ 2 callersClassPalette
models/utils/distance_util.py:24
↓ 2 callersClassPointnetFPModule
r"""Propigates the features of one set to another Parameters ---------- mlp : list Pointnet module parameters bn : bool
pointnet2/pointnet2_modules.py:356
↓ 2 callersClassPositionEmbeddingLearned
Absolute pos embedding, learned.
models/pq_transformer.py:17
↓ 2 callersClassPredictHead
models/pq_transformer.py:60
↓ 2 callersClassQuadPredictHead
models/pq_transformer.py:94
↓ 2 callersClassTransformerDecoderLayer
models/transformer.py:162
↓ 2 callersClassVotingModule
models/voting_module.py:16
↓ 1 callersClassConv2d
pointnet2/pytorch_utils.py:157
↓ 1 callersClassDoubleLoader
double_loader.py:1
↓ 1 callersClassFPSModule
models/utils/pointnet_util.py:52
↓ 1 callersClassGradualWarmupScheduler
Gradually warm-up(increasing) learning rate in optimizer. Proposed in 'Accurate, Large Minibatch SGD: Training ImageNet in 1 Hour'. Args:
utils/lr_scheduler.py:6
↓ 1 callersClassPaletteColor
models/utils/distance_util.py:9
↓ 1 callersClassPointnet2Backbone
r""" Backbone network for point cloud feature learning. Based on Pointnet++ single-scale grouping network. Parameters
models/backbone_module.py:21
↓ 1 callersClassPointnetSAModuleMSG
r"""Pointnet set abstrction layer with multiscale grouping Parameters ---------- npoint : int Number of features radii : list
pointnet2/pointnet2_modules.py:78
↓ 1 callersClassTransformer
models/transformer.py:16
↓ 1 callersClassTransformerDecoder
models/transformer.py:80
↓ 1 callersClassTransformerEncoder
models/transformer.py:56
↓ 1 callersClassTransformerEncoderLayer
models/transformer.py:115
↓ 1 callersClass_ColorfulFormatter
utils/logger.py:9
ClassBNMomentumScheduler
pointnet2/pytorch_utils.py:271
ClassBallQuery
pointnet2/pointnet2_utils.py:260
ClassBatchNorm2d
pointnet2/pytorch_utils.py:55
ClassBatchNorm3d
pointnet2/pytorch_utils.py:61
ClassConv1d
pointnet2/pytorch_utils.py:123
ClassConv3d
pointnet2/pytorch_utils.py:191
ClassFC
pointnet2/pytorch_utils.py:225
ClassFurthestPointSampling
pointnet2/pointnet2_utils.py:51
ClassGammaDistribution
fit.py:30
ClassGatherOperation
pointnet2/pointnet2_utils.py:83
ClassGeneralSamplingModule
models/utils/pointnet_util.py:363
ClassGroupAll
r""" Groups all features Parameters ---------
pointnet2/pointnet2_utils.py:379
ClassGroupingOperation
pointnet2/pointnet2_utils.py:209
ClassLogger
utils/tf_logger.py:15
ClassPDFunction
fit.py:14
ClassPointNetFeaturePropagation
models/utils/pointnet_util.py:279
ClassPointNetSetAbstraction
models/utils/pointnet_util.py:177
ClassPointNetSetAbstractionMsg
models/utils/pointnet_util.py:217
ClassPointnetLFPModuleMSG
Modified based on _PointnetSAModuleBase and PointnetSAModuleMSG learnable feature propagation layer.
pointnet2/pointnet2_modules.py:418
ClassPointnetSAModule
r"""Pointnet set abstrction layer Parameters ---------- npoint : int Number of features radius : float Radius of ball
pointnet2/pointnet2_modules.py:127
ClassPointnetSAModuleMSGVotes
Modified based on _PointnetSAModuleBase and PointnetSAModuleMSG with extra support for returning point indices for getting their GT votes
pointnet2/pointnet2_modules.py:274
ClassPointsObjClsModule
models/utils/pointnet_util.py:333
ClassPoissonDistribution
fit.py:49
ClassQueryAndGroup
r""" Groups with a ball query of radius Parameters --------- radius : float32 Radius of ball nsample : int32 Maxi
pointnet2/pointnet2_utils.py:294
ClassRandomDropout
pointnet2/pointnet2_utils.py:40
ClassSharedMLP
pointnet2/pytorch_utils.py:11
ClassSigmoidFocalClassificationLoss
Sigmoid focal cross entropy loss.
models/utils/losses.py:21
ClassThreeInterpolate
pointnet2/pointnet2_utils.py:152
ClassThreeNN
pointnet2/pointnet2_utils.py:120
ClassVisualizer
utils/tf_visualizer.py:15
Class_BNBase
pointnet2/pytorch_utils.py:39
Class_ConvBase
pointnet2/pytorch_utils.py:67
Class_PointnetSAModuleBase
pointnet2/pointnet2_modules.py:26