Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/AIR-DISCOVER/Omni-PQ
/ types & classes
Types & classes
63 in github.com/AIR-DISCOVER/Omni-PQ
⨍
Functions
370
◇
Types & classes
63
↓ 7 callers
Class
ScannetDatasetConfig
scannet/model_util_scannet.py:14
↓ 6 callers
Class
ScannetDetectionDataset
scannet/scannet_detection_dataset.py:36
↓ 5 callers
Class
PointnetSAModuleVotes
Modified based on _PointnetSAModuleBase and PointnetSAModuleMSG with extra support for returning point indices for getting their GT votes
pointnet2/pointnet2_modules.py:164
↓ 4 callers
Class
ARKitSceneDataset
ARKitScenes/arkitscenes_dataset.py:42
↓ 3 callers
Class
MultiheadAttention
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 callers
Class
QUADAPCalculator
Calculating Average Precision
models/ap_helper_pq.py:579
↓ 2 callers
Class
APCalculator
Calculating Average Precision
models/ap_helper_pq.py:520
↓ 2 callers
Class
BatchNorm1d
pointnet2/pytorch_utils.py:49
↓ 2 callers
Class
DoubleLoaderTest
double_loader.py:32
↓ 2 callers
Class
FitRunner
fit.py:70
↓ 2 callers
Class
PQ_Transformer
models/pq_transformer.py:123
↓ 2 callers
Class
Palette
models/utils/distance_util.py:24
↓ 2 callers
Class
PointnetFPModule
r"""Propigates the features of one set to another Parameters ---------- mlp : list Pointnet module parameters bn : bool
pointnet2/pointnet2_modules.py:356
↓ 2 callers
Class
PositionEmbeddingLearned
Absolute pos embedding, learned.
models/pq_transformer.py:17
↓ 2 callers
Class
PredictHead
models/pq_transformer.py:60
↓ 2 callers
Class
QuadPredictHead
models/pq_transformer.py:94
↓ 2 callers
Class
TransformerDecoderLayer
models/transformer.py:162
↓ 2 callers
Class
VotingModule
models/voting_module.py:16
↓ 1 callers
Class
Conv2d
pointnet2/pytorch_utils.py:157
↓ 1 callers
Class
DoubleLoader
double_loader.py:1
↓ 1 callers
Class
FPSModule
models/utils/pointnet_util.py:52
↓ 1 callers
Class
GradualWarmupScheduler
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 callers
Class
PaletteColor
models/utils/distance_util.py:9
↓ 1 callers
Class
Pointnet2Backbone
r""" Backbone network for point cloud feature learning. Based on Pointnet++ single-scale grouping network. Parameters
models/backbone_module.py:21
↓ 1 callers
Class
PointnetSAModuleMSG
r"""Pointnet set abstrction layer with multiscale grouping Parameters ---------- npoint : int Number of features radii : list
pointnet2/pointnet2_modules.py:78
↓ 1 callers
Class
Transformer
models/transformer.py:16
↓ 1 callers
Class
TransformerDecoder
models/transformer.py:80
↓ 1 callers
Class
TransformerEncoder
models/transformer.py:56
↓ 1 callers
Class
TransformerEncoderLayer
models/transformer.py:115
↓ 1 callers
Class
_ColorfulFormatter
utils/logger.py:9
Class
BNMomentumScheduler
pointnet2/pytorch_utils.py:271
Class
BallQuery
pointnet2/pointnet2_utils.py:260
Class
BatchNorm2d
pointnet2/pytorch_utils.py:55
Class
BatchNorm3d
pointnet2/pytorch_utils.py:61
Class
Conv1d
pointnet2/pytorch_utils.py:123
Class
Conv3d
pointnet2/pytorch_utils.py:191
Class
FC
pointnet2/pytorch_utils.py:225
Class
FurthestPointSampling
pointnet2/pointnet2_utils.py:51
Class
GammaDistribution
fit.py:30
Class
GatherOperation
pointnet2/pointnet2_utils.py:83
Class
GeneralSamplingModule
models/utils/pointnet_util.py:363
Class
GroupAll
r""" Groups all features Parameters ---------
pointnet2/pointnet2_utils.py:379
Class
GroupingOperation
pointnet2/pointnet2_utils.py:209
Class
Logger
utils/tf_logger.py:15
Class
PDFunction
fit.py:14
Class
PointNetFeaturePropagation
models/utils/pointnet_util.py:279
Class
PointNetSetAbstraction
models/utils/pointnet_util.py:177
Class
PointNetSetAbstractionMsg
models/utils/pointnet_util.py:217
Class
PointnetLFPModuleMSG
Modified based on _PointnetSAModuleBase and PointnetSAModuleMSG learnable feature propagation layer.
pointnet2/pointnet2_modules.py:418
Class
PointnetSAModule
r"""Pointnet set abstrction layer Parameters ---------- npoint : int Number of features radius : float Radius of ball
pointnet2/pointnet2_modules.py:127
Class
PointnetSAModuleMSGVotes
Modified based on _PointnetSAModuleBase and PointnetSAModuleMSG with extra support for returning point indices for getting their GT votes
pointnet2/pointnet2_modules.py:274
Class
PointsObjClsModule
models/utils/pointnet_util.py:333
Class
PoissonDistribution
fit.py:49
Class
QueryAndGroup
r""" Groups with a ball query of radius Parameters --------- radius : float32 Radius of ball nsample : int32 Maxi
pointnet2/pointnet2_utils.py:294
Class
RandomDropout
pointnet2/pointnet2_utils.py:40
Class
SharedMLP
pointnet2/pytorch_utils.py:11
Class
SigmoidFocalClassificationLoss
Sigmoid focal cross entropy loss.
models/utils/losses.py:21
Class
ThreeInterpolate
pointnet2/pointnet2_utils.py:152
Class
ThreeNN
pointnet2/pointnet2_utils.py:120
Class
Visualizer
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