Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/HKUST-Aerial-Robotics/SG-Reg
/ types & classes
Types & classes
86 in github.com/HKUST-Aerial-Robotics/SG-Reg
⨍
Functions
402
◇
Types & classes
86
↓ 11 callers
Class
ResidualBlock
sgreg/kpconv/modules.py:149
↓ 7 callers
Class
PointXYZ
sgreg/extensions/extra/cloud/cloud.h:14
↓ 6 callers
Class
TicToc
sgreg/utils/tictoc.py:3
↓ 5 callers
Class
SearchParams
sgreg/extensions/extra/nanoflann/nanoflann.hpp:554
↓ 4 callers
Class
KDTreeSingleIndexAdaptorParams
sgreg/extensions/extra/nanoflann/nanoflann.hpp:546
↓ 4 callers
Class
ScenePairDataset
sgreg/dataset/scene_pair_dataset.py:77
↓ 4 callers
Class
UnaryBlock
sgreg/kpconv/modules.py:53
↓ 3 callers
Class
Attention
sgreg/gnn/spatial_attention.py:45
↓ 3 callers
Class
Evaluator
sgreg/loss/eval.py:8
↓ 3 callers
Class
GroupNorm
sgreg/kpconv/modules.py:33
↓ 3 callers
Class
IndexDist_Sorter
sgreg/extensions/extra/nanoflann/nanoflann.hpp:208
↓ 3 callers
Class
TripletGNN
sgreg/gnn/triplet_gnn.py:266
↓ 2 callers
Class
GraphNeuralNetwork
sgreg/gnn/gnn.py:117
↓ 2 callers
Class
Instance
sgreg/dataset/scene_graph.py:11
↓ 2 callers
Class
KPConv
sgreg/kpconv/kpconv.py:10
↓ 2 callers
Class
KPConvFPN
sgreg/backbone/backbone.py:42
↓ 2 callers
Class
LastUnaryBlock
sgreg/kpconv/modules.py:86
↓ 2 callers
Class
LearnableLogOptimalTransport
Adopted from https://github.com/qinzheng93/GeoTransformer
sgreg/match/learnable_sinkhorn.py:5
↓ 2 callers
Class
RecallMetrics
sgreg/utils/utils.py:8
↓ 2 callers
Class
SEModule
sgreg/backbone/senet.py:6
↓ 2 callers
Class
SGNet
sgreg/sg_reg.py:23
↓ 2 callers
Class
SelfBlock
sgreg/gnn/spatial_attention.py:88
↓ 2 callers
Class
SinusoidalPositionalEmbedding
sgreg/gnn/triplet_gnn.py:204
↓ 2 callers
Class
SpatialTransformer
sgreg/gnn/spatial_attention.py:210
↓ 1 callers
Class
ConvBlock
sgreg/kpconv/modules.py:104
↓ 1 callers
Class
CrossBlock
sgreg/gnn/spatial_attention.py:151
↓ 1 callers
Class
FineMatchingLoss
sgreg/loss/loss.py:242
↓ 1 callers
Class
HybridReg
sgreg/registration/hybrid_reg.py:9
↓ 1 callers
Class
InstanceMatchingLoss
sgreg/loss/loss.py:311
↓ 1 callers
Class
KPConvShape
sgreg/backbone/shape_encoder.py:8
↓ 1 callers
Class
LearnableFourierPositionalEncoding
sgreg/gnn/spatial_attention.py:24
↓ 1 callers
Class
LocalGlobalRegistration
sgreg/registration/local_global_registration.py:11
↓ 1 callers
Class
MatchAssignment
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 callers
Class
NodesInitLayer
Initialize semantic, boundingbox and shape embeddings of each node. Output fused node embeddings.
sgreg/gnn/nodes_init_layer.py:6
↓ 1 callers
Class
PointGCN
sgreg/gnn/gnn.py:30
↓ 1 callers
Class
SampledData
sgreg/extensions/cpu/grid_subsampling/grid_subsampling_cpu.h:7
↓ 1 callers
Class
SampledInst
sgreg/extensions/cpu/grid_subsampling/grid_subsampling_cpu.h:23
↓ 1 callers
Class
SelfGAT
sgreg/gnn/gnn.py:41
↓ 1 callers
Class
SinkhornMatch
sgreg/match/match.py:227
↓ 1 callers
Class
SinusoidalPositionalEmbedding2
sgreg/gnn/triplet_gnn.py:237
↓ 1 callers
Class
TicToc
sgreg/utils/utils.py:37
↓ 1 callers
Class
WarmUpCosineAnnealingFunction
sgreg/utils/torch.py:277
↓ 1 callers
Class
WeightedProcrustes
sgreg/registration/procrustes.py:76
Class
BertModelWarper
sgreg/bert/bertwarper.py:14
Class
CosineAnnealingFunction
sgreg/utils/torch.py:267
Class
CrossGraphLayer
sgreg/gnn/gnn.py:75
Class
GlobalAvgPool
sgreg/kpconv/modules.py:27
Class
Interval
sgreg/extensions/extra/nanoflann/nanoflann.hpp:771
Class
KDTreeBaseClass
sgreg/extensions/extra/nanoflann/nanoflann.hpp:739
Class
KDTreeEigenMatrixAdaptor
sgreg/extensions/extra/nanoflann/nanoflann.hpp:1957
Class
KDTreeSingleIndexAdaptor
sgreg/extensions/extra/nanoflann/nanoflann.hpp:1116
Class
KDTreeSingleIndexDynamicAdaptor
sgreg/extensions/extra/nanoflann/nanoflann.hpp:1791
Class
KDTreeSingleIndexDynamicAdaptor_
sgreg/extensions/extra/nanoflann/nanoflann.hpp:1468
Class
KNNInterpolate
sgreg/kpconv/modules.py:8
Class
KNNResultSet
sgreg/extensions/extra/nanoflann/nanoflann.hpp:142
Class
L1_Adaptor
sgreg/extensions/extra/nanoflann/nanoflann.hpp:324
Class
L2_Adaptor
sgreg/extensions/extra/nanoflann/nanoflann.hpp:375
Class
L2_Simple_Adaptor
sgreg/extensions/extra/nanoflann/nanoflann.hpp:423
Class
MaxPool
sgreg/kpconv/modules.py:21
Class
Metric
sgreg/extensions/extra/nanoflann/nanoflann.hpp:315
Class
Node
sgreg/extensions/extra/nanoflann/nanoflann.hpp:754
Class
PointCloud
sgreg/extensions/extra/cloud/cloud.h:108
Class
PooledAllocator
sgreg/extensions/extra/nanoflann/nanoflann.hpp:601
Class
RadiusResultSet
sgreg/extensions/extra/nanoflann/nanoflann.hpp:220
Class
SO2_Adaptor
sgreg/extensions/extra/nanoflann/nanoflann.hpp:455
Class
SO3_Adaptor
sgreg/extensions/extra/nanoflann/nanoflann.hpp:489
Class
TextEncoderShell
sgreg/bert/bertwarper.py:166
Class
TmpKPConvFPN
sgreg/backbone/backbone.py:129
Class
array_or_vector_selector
sgreg/extensions/extra/nanoflann/nanoflann.hpp:715
Class
array_or_vector_selector<-1, T>
sgreg/extensions/extra/nanoflann/nanoflann.hpp:719
Class
has_assign
sgreg/extensions/extra/nanoflann/nanoflann.hpp:93
Class
has_assign<T, decltype((void)std::declval<T>().assign(1, 0), 0)>
sgreg/extensions/extra/nanoflann/nanoflann.hpp:96
Class
has_resize
sgreg/extensions/extra/nanoflann/nanoflann.hpp:87
Class
has_resize<T, decltype((void)std::declval<T>().resize(1), 0)>
sgreg/extensions/extra/nanoflann/nanoflann.hpp:90
Class
leaf
sgreg/extensions/extra/nanoflann/nanoflann.hpp:758
Class
metric_L1
sgreg/extensions/extra/nanoflann/nanoflann.hpp:510
Class
metric_L2
sgreg/extensions/extra/nanoflann/nanoflann.hpp:516
Class
metric_L2_Simple
sgreg/extensions/extra/nanoflann/nanoflann.hpp:522
Class
metric_SO2
sgreg/extensions/extra/nanoflann/nanoflann.hpp:528
Class
metric_SO3
sgreg/extensions/extra/nanoflann/nanoflann.hpp:534
Class
nonleaf
sgreg/extensions/extra/nanoflann/nanoflann.hpp:761
Class
traits
sgreg/extensions/extra/nanoflann/nanoflann.hpp:511
Class
traits
sgreg/extensions/extra/nanoflann/nanoflann.hpp:517
Class
traits
sgreg/extensions/extra/nanoflann/nanoflann.hpp:523
Class
traits
sgreg/extensions/extra/nanoflann/nanoflann.hpp:529
Class
traits
sgreg/extensions/extra/nanoflann/nanoflann.hpp:535