MCPcopy Create free account

hub / github.com/HKUST-Aerial-Robotics/SG-Reg / types & classes

Types & classes86 in github.com/HKUST-Aerial-Robotics/SG-Reg

↓ 11 callersClassResidualBlock
sgreg/kpconv/modules.py:149
↓ 7 callersClassPointXYZ
sgreg/extensions/extra/cloud/cloud.h:14
↓ 6 callersClassTicToc
sgreg/utils/tictoc.py:3
↓ 5 callersClassSearchParams
sgreg/extensions/extra/nanoflann/nanoflann.hpp:554
↓ 4 callersClassKDTreeSingleIndexAdaptorParams
sgreg/extensions/extra/nanoflann/nanoflann.hpp:546
↓ 4 callersClassScenePairDataset
sgreg/dataset/scene_pair_dataset.py:77
↓ 4 callersClassUnaryBlock
sgreg/kpconv/modules.py:53
↓ 3 callersClassAttention
sgreg/gnn/spatial_attention.py:45
↓ 3 callersClassEvaluator
sgreg/loss/eval.py:8
↓ 3 callersClassGroupNorm
sgreg/kpconv/modules.py:33
↓ 3 callersClassIndexDist_Sorter
sgreg/extensions/extra/nanoflann/nanoflann.hpp:208
↓ 3 callersClassTripletGNN
sgreg/gnn/triplet_gnn.py:266
↓ 2 callersClassGraphNeuralNetwork
sgreg/gnn/gnn.py:117
↓ 2 callersClassInstance
sgreg/dataset/scene_graph.py:11
↓ 2 callersClassKPConv
sgreg/kpconv/kpconv.py:10
↓ 2 callersClassKPConvFPN
sgreg/backbone/backbone.py:42
↓ 2 callersClassLastUnaryBlock
sgreg/kpconv/modules.py:86
↓ 2 callersClassLearnableLogOptimalTransport
Adopted from https://github.com/qinzheng93/GeoTransformer
sgreg/match/learnable_sinkhorn.py:5
↓ 2 callersClassRecallMetrics
sgreg/utils/utils.py:8
↓ 2 callersClassSEModule
sgreg/backbone/senet.py:6
↓ 2 callersClassSGNet
sgreg/sg_reg.py:23
↓ 2 callersClassSelfBlock
sgreg/gnn/spatial_attention.py:88
↓ 2 callersClassSinusoidalPositionalEmbedding
sgreg/gnn/triplet_gnn.py:204
↓ 2 callersClassSpatialTransformer
sgreg/gnn/spatial_attention.py:210
↓ 1 callersClassConvBlock
sgreg/kpconv/modules.py:104
↓ 1 callersClassCrossBlock
sgreg/gnn/spatial_attention.py:151
↓ 1 callersClassFineMatchingLoss
sgreg/loss/loss.py:242
↓ 1 callersClassHybridReg
sgreg/registration/hybrid_reg.py:9
↓ 1 callersClassInstanceMatchingLoss
sgreg/loss/loss.py:311
↓ 1 callersClassKPConvShape
sgreg/backbone/shape_encoder.py:8
↓ 1 callersClassLearnableFourierPositionalEncoding
sgreg/gnn/spatial_attention.py:24
↓ 1 callersClassLocalGlobalRegistration
sgreg/registration/local_global_registration.py:11
↓ 1 callersClassMatchAssignment
This is a module adopted from LightGlue. It can be used: scores, sim = MatchAssignment(dim)(desc0, desc1), desc0, desc1: (b,
sgreg/match/match.py:122
↓ 1 callersClassNodesInitLayer
Initialize semantic, boundingbox and shape embeddings of each node. Output fused node embeddings.
sgreg/gnn/nodes_init_layer.py:6
↓ 1 callersClassPointGCN
sgreg/gnn/gnn.py:30
↓ 1 callersClassSampledData
sgreg/extensions/cpu/grid_subsampling/grid_subsampling_cpu.h:7
↓ 1 callersClassSampledInst
sgreg/extensions/cpu/grid_subsampling/grid_subsampling_cpu.h:23
↓ 1 callersClassSelfGAT
sgreg/gnn/gnn.py:41
↓ 1 callersClassSinkhornMatch
sgreg/match/match.py:227
↓ 1 callersClassSinusoidalPositionalEmbedding2
sgreg/gnn/triplet_gnn.py:237
↓ 1 callersClassTicToc
sgreg/utils/utils.py:37
↓ 1 callersClassWarmUpCosineAnnealingFunction
sgreg/utils/torch.py:277
↓ 1 callersClassWeightedProcrustes
sgreg/registration/procrustes.py:76
ClassBertModelWarper
sgreg/bert/bertwarper.py:14
ClassCosineAnnealingFunction
sgreg/utils/torch.py:267
ClassCrossGraphLayer
sgreg/gnn/gnn.py:75
ClassGlobalAvgPool
sgreg/kpconv/modules.py:27
ClassInterval
sgreg/extensions/extra/nanoflann/nanoflann.hpp:771
ClassKDTreeBaseClass
sgreg/extensions/extra/nanoflann/nanoflann.hpp:739
ClassKDTreeEigenMatrixAdaptor
sgreg/extensions/extra/nanoflann/nanoflann.hpp:1957
ClassKDTreeSingleIndexAdaptor
sgreg/extensions/extra/nanoflann/nanoflann.hpp:1116
ClassKDTreeSingleIndexDynamicAdaptor
sgreg/extensions/extra/nanoflann/nanoflann.hpp:1791
ClassKDTreeSingleIndexDynamicAdaptor_
sgreg/extensions/extra/nanoflann/nanoflann.hpp:1468
ClassKNNInterpolate
sgreg/kpconv/modules.py:8
ClassKNNResultSet
sgreg/extensions/extra/nanoflann/nanoflann.hpp:142
ClassL1_Adaptor
sgreg/extensions/extra/nanoflann/nanoflann.hpp:324
ClassL2_Adaptor
sgreg/extensions/extra/nanoflann/nanoflann.hpp:375
ClassL2_Simple_Adaptor
sgreg/extensions/extra/nanoflann/nanoflann.hpp:423
ClassMaxPool
sgreg/kpconv/modules.py:21
ClassMetric
sgreg/extensions/extra/nanoflann/nanoflann.hpp:315
ClassNode
sgreg/extensions/extra/nanoflann/nanoflann.hpp:754
ClassPointCloud
sgreg/extensions/extra/cloud/cloud.h:108
ClassPooledAllocator
sgreg/extensions/extra/nanoflann/nanoflann.hpp:601
ClassRadiusResultSet
sgreg/extensions/extra/nanoflann/nanoflann.hpp:220
ClassSO2_Adaptor
sgreg/extensions/extra/nanoflann/nanoflann.hpp:455
ClassSO3_Adaptor
sgreg/extensions/extra/nanoflann/nanoflann.hpp:489
ClassTextEncoderShell
sgreg/bert/bertwarper.py:166
ClassTmpKPConvFPN
sgreg/backbone/backbone.py:129
Classarray_or_vector_selector
sgreg/extensions/extra/nanoflann/nanoflann.hpp:715
Classarray_or_vector_selector<-1, T>
sgreg/extensions/extra/nanoflann/nanoflann.hpp:719
Classhas_assign
sgreg/extensions/extra/nanoflann/nanoflann.hpp:93
Classhas_assign<T, decltype((void)std::declval<T>().assign(1, 0), 0)>
sgreg/extensions/extra/nanoflann/nanoflann.hpp:96
Classhas_resize
sgreg/extensions/extra/nanoflann/nanoflann.hpp:87
Classhas_resize<T, decltype((void)std::declval<T>().resize(1), 0)>
sgreg/extensions/extra/nanoflann/nanoflann.hpp:90
Classleaf
sgreg/extensions/extra/nanoflann/nanoflann.hpp:758
Classmetric_L1
sgreg/extensions/extra/nanoflann/nanoflann.hpp:510
Classmetric_L2
sgreg/extensions/extra/nanoflann/nanoflann.hpp:516
Classmetric_L2_Simple
sgreg/extensions/extra/nanoflann/nanoflann.hpp:522
Classmetric_SO2
sgreg/extensions/extra/nanoflann/nanoflann.hpp:528
Classmetric_SO3
sgreg/extensions/extra/nanoflann/nanoflann.hpp:534
Classnonleaf
sgreg/extensions/extra/nanoflann/nanoflann.hpp:761
Classtraits
sgreg/extensions/extra/nanoflann/nanoflann.hpp:511
Classtraits
sgreg/extensions/extra/nanoflann/nanoflann.hpp:517
Classtraits
sgreg/extensions/extra/nanoflann/nanoflann.hpp:523
Classtraits
sgreg/extensions/extra/nanoflann/nanoflann.hpp:529
Classtraits
sgreg/extensions/extra/nanoflann/nanoflann.hpp:535