Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/HpWang-whu/RoReg
/ functions
Functions
1,370 in github.com/HpWang-whu/RoReg
⨍
Functions
1,370
◇
Types & classes
420
↓ 2 callers
Function
load_model
Creates and returns an instance of the model given its class name.
backbone/fcgf/__init__.py:16
↓ 2 callers
Function
make_data_loader
(phase, augment_data, batch_size, shuffle, num_workers, repeat, config)
utils/MinkowskiEngine/examples/vae.py:156
↓ 2 callers
Method
network_initialization
(self, in_channels, out_channels, D)
utils/MinkowskiEngine/examples/resnet.py:67
↓ 2 callers
Method
nodes
utils/MinkowskiEngine/src/3rdparty/robin_hood.h:1135
↓ 2 callers
Function
not_reentrant_error
(error_str='')
utils/MinkowskiEngine/MinkowskiEngine/utils/gradcheck.py:196
↓ 2 callers
Function
ones
utils/MinkowskiEngine/src/coordinate_map_manager.cpp:45
↓ 2 callers
Method
origin
(self)
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiCoordinateManager.py:270
↓ 2 callers
Method
origin
utils/MinkowskiEngine/src/coordinate_map_manager.cpp:470
↓ 2 callers
Method
ransac
(self,dataset,keynum,max_iter=1000)
test/estimator.py:245
↓ 2 callers
Function
read
(*parts)
utils/MinkowskiEngine/setup.py:76
↓ 2 callers
Method
recall
(self,data)
train/val.py:13
↓ 2 callers
Function
resample_mesh
https://chrischoy.github.io/research/barycentric-coordinate-for-mesh-sampling/ Samples point cloud on the surface of the model defined as vec
utils/MinkowskiEngine/examples/reconstruction.py:81
↓ 2 callers
Method
reset_permutation
(self)
utils/MinkowskiEngine/examples/reconstruction.py:63
↓ 2 callers
Method
reset_permutation
(self)
utils/MinkowskiEngine/examples/common.py:76
↓ 2 callers
Function
run_command
(*args)
utils/MinkowskiEngine/setup.py:89
↓ 2 callers
Function
save_ctx
( ctx, # function object context kernel_generator: KernelGenerator, in_coords_key: CoordinateMapK
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiKernelGenerator.py:383
↓ 2 callers
Function
score_mat
(source,target)
network/rot_coh_match.py:8
↓ 2 callers
Function
set_global_coordinate_manager
r"""Set the global coordinate manager. :attr:`MinkowskiEngine.CoordinateManager` The coordinate manager which will be set to the global coord
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiTensor.py:117
↓ 2 callers
Function
set_sparse_tensor_operation_mode
r"""Define the sparse tensor coordinate manager operation mode. By default, a :attr:`MinkowskiEngine.SparseTensor.SparseTensor` instantiation
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiTensor.py:68
↓ 2 callers
Method
size
utils/MinkowskiEngine/src/3rdparty/robin_hood.h:1878
↓ 2 callers
Method
size
Returns the size of the array.
utils/MinkowskiEngine/src/primitives/small_vector.hpp:193
↓ 2 callers
Method
slice
r""" Args: :attr:`X` (:attr:`MinkowskiEngine.SparseTensor`): a sparse tensor that discretized the original input.
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiSparseTensor.py:530
↓ 2 callers
Function
sparse_collate
r"""Create input arguments for a sparse tensor `the documentation <https://nvidia.github.io/MinkowskiEngine/sparse_tensor.html>`_. Convert a
utils/MinkowskiEngine/MinkowskiEngine/utils/collation.py:96
↓ 2 callers
Function
sparse_tensor_operation_mode
r"""Return the current sparse tensor operation mode.
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiTensor.py:103
↓ 2 callers
Method
stride_map
(self, in_key: CoordinateMapKey, stride_key: CoordinateMapKey)
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiCoordinateManager.py:429
↓ 2 callers
Function
swap
utils/MinkowskiEngine/src/primitives/small_vector.hpp:331
↓ 2 callers
Function
test
(net, device, config, phase="val")
utils/MinkowskiEngine/examples/classification_modelnet40.py:238
↓ 2 callers
Method
tic
(self)
utils/MinkowskiEngine/examples/common.py:46
↓ 2 callers
Function
to_cuda
(data)
utils/utils.py:104
↓ 2 callers
Method
toc
(self, average=False)
utils/MinkowskiEngine/examples/common.py:51
↓ 2 callers
Method
union_map
(self, in_keys: list, out_key)
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiCoordinateManager.py:432
↓ 2 callers
Function
warn_bc_breaking
()
utils/MinkowskiEngine/MinkowskiEngine/utils/gradcheck.py:239
↓ 1 callers
Method
Batch_Des2R_torch
(self,des1_eqv,des2_eqv)
test/estimator.py:85
↓ 1 callers
Method
Clean_cache
(self)
train/trainset/GF_ET.py:291
↓ 1 callers
Method
DR_statictic
(self,DR_indexs)
test/estimator.py:119
↓ 1 callers
Method
DeltaR
(self,R,index)
dataops/dataloader.py:195
↓ 1 callers
Method
DeltaR
(self,R,index)
train/trainset/GF_ET.py:156
↓ 1 callers
Method
Des2DR
(self,Des1,Des2)
network/group_feat.py:55
↓ 1 callers
Method
FCGF_Group_Feature_Extractor
(self,Point,Keys_index)
train/trainset/GF_ET.py:70
↓ 1 callers
Method
Feature_extracting
(self, data_loader)
testset.py:124
↓ 1 callers
Method
Gen_trainset
(self)
train/trainset/GF_ET.py:163
↓ 1 callers
Method
Gen_valset
(self)
train/trainset/GF_ET.py:250
↓ 1 callers
Method
PartII_SO3_Conv
(self,data)
network/eqv_trans.py:39
↓ 1 callers
Method
PartII_SO3_Conv
(self,data)
network/eqv_trans.py:111
↓ 1 callers
Method
R2DR_id
(self,R)
dataops/dataloader.py:147
↓ 1 callers
Method
R2DR_id
(self,R)
train/trainset/RM.py:162
↓ 1 callers
Method
R_cal
(self,key_m0,key_m1,center0,center1,scores)
test/estimator.py:45
↓ 1 callers
Method
Rargu_on1
(self,keys1,feats1)
train/trainset/RM.py:154
↓ 1 callers
Method
Rt_cal
(self,key_m0,key_m1,scores)
test/estimator.py:53
↓ 1 callers
Method
SO3_Conv
(self,data)
network/group_feat.py:26
↓ 1 callers
Method
SVDR_w
(self,beforerot,afterrot,scores)
test/estimator.py:39
↓ 1 callers
Method
TV_RM
(self,batchnum=30000)
train/trainset/RM.py:90
↓ 1 callers
Function
Threepps2Tran
(kps0_init,kps1_init)
utils/utils.py:242
↓ 1 callers
Method
Threepps2Tran
(self,kps0_init,kps1_init)
test/estimator.py:139
↓ 1 callers
Method
__init__
(self, log_dir)
utils/utils.py:201
↓ 1 callers
Method
__init__
(self, in_channel, out_channel)
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiNonlinearity.py:176
↓ 1 callers
Method
__init__
(self, inplanes, planes, stride=1, dilatio
utils/MinkowskiEngine/MinkowskiEngine/modules/resnet_block.py:32
↓ 1 callers
Method
__init__
(self, in_channels, out_channels, D=3)
utils/MinkowskiEngine/examples/multigpu_lightning.py:79
↓ 1 callers
Method
__init__
(self, data_source, shuffle=False)
utils/MinkowskiEngine/examples/reconstruction.py:58
↓ 1 callers
Method
__init__
(self, cfg)
checkpoints/FCGF/RD_former/rot_detect.py:7
↓ 1 callers
Method
__init__
(self,in_dim,out_dim)
network/ops.py:12
↓ 1 callers
Method
__init__
(self, cfg)
network/rot_detect.py:9
↓ 1 callers
Method
__init__
(self, cfg)
network/eqv_trans.py:79
↓ 1 callers
Function
_calculate_correct_fan
(tensor, mode)
utils/MinkowskiEngine/MinkowskiEngine/utils/init.py:25
↓ 1 callers
Function
_calculate_fan_in_and_fan_out
(tensor)
utils/MinkowskiEngine/MinkowskiEngine/utils/init.py:5
↓ 1 callers
Method
_get_coordinate_field
(self)
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiTensorField.py:251
↓ 1 callers
Method
_get_coordinates
(self)
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiTensor.py:225
↓ 1 callers
Method
_init_dataset
(self)
train/trainer.py:53
↓ 1 callers
Method
_init_logger
(self)
train/trainer.py:86
↓ 1 callers
Method
_init_network
(self)
train/trainer.py:61
↓ 1 callers
Method
_is_same_key
(self, other)
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiTensorField.py:370
↓ 1 callers
Method
_load_model
(self)
test/extractor.py:22
↓ 1 callers
Method
_load_model
(self)
test/detector.py:19
↓ 1 callers
Method
_load_model
(self)
test/matcher.py:121
↓ 1 callers
Method
_load_model
(self)
test/estimator.py:284
↓ 1 callers
Method
_load_model
(self)
train/trainer.py:68
↓ 1 callers
Method
_sample_noise
(self, num, noise_params)
utils/MinkowskiEngine/examples/training.py:109
↓ 1 callers
Method
_sample_xs
Return random numbers between line_params[0], line_params[1]
utils/MinkowskiEngine/examples/training.py:113
↓ 1 callers
Function
_sum
r"""Compute the sum of sparse tensor features Sum all sparse tensor features. All sparse tensors must have the same `coordinate_map_key` (the
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiOps.py:161
↓ 1 callers
Method
_uniform_to_angle
(self, u)
utils/MinkowskiEngine/examples/training.py:105
↓ 1 callers
Method
allocate
clang-format off
utils/MinkowskiEngine/src/coordinate_map.hpp:184
↓ 1 callers
Method
allocate
allocates, but does NOT initialize. Use in-place new constructor, e.g. T* obj = pool.allocate(); ::new (static_cast<void*>(obj)) T();
utils/MinkowskiEngine/src/3rdparty/robin_hood.h:415
↓ 1 callers
Function
attention
(query, key, value)
network/rot_coh_match.py:84
↓ 1 callers
Method
batch_create
(self,feats0_fcgf,feats1_fcgf,feats0_yomo,feats1_yomo,index_pre,start,end)
test/estimator.py:293
↓ 1 callers
Method
batch_feature_extraction
(self)
testset.py:184
↓ 1 callers
Method
begin
utils/MinkowskiEngine/src/3rdparty/robin_hood.h:1756
↓ 1 callers
Method
calcNumBytesInfo
utils/MinkowskiEngine/src/3rdparty/robin_hood.h:1923
↓ 1 callers
Method
calcNumBytesTotal
utils/MinkowskiEngine/src/3rdparty/robin_hood.h:1936
↓ 1 callers
Function
cat
r"""Concatenate sparse tensors Concatenate sparse tensor features. All sparse tensors must have the same `coordinate_map_key` (the same coord
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiOps.py:141
↓ 1 callers
Method
cend
utils/MinkowskiEngine/src/3rdparty/robin_hood.h:1806
↓ 1 callers
Method
center_cal
(self,key_m0,key_m1,scores)
test/estimator.py:32
↓ 1 callers
Function
check_undefined_grad_support
(output_to_check)
utils/MinkowskiEngine/MinkowskiEngine/utils/gradcheck.py:246
↓ 1 callers
Function
clear_global_coordinate_manager
r"""Clear the global coordinate manager cache. When you use the operation mode: :attr:`MinkowskiEngine.SparseTensor.SparseTensorOperationMode
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiTensor.py:127
↓ 1 callers
Function
conv_on_coords
()
utils/MinkowskiEngine/examples/convolution.py:93
↓ 1 callers
Function
convert_region_type
when center is True, the custom region_offset will be centered at the origin. Currently, for HYPER_CUBE, HYPER_CROSS with odd kernel sizes ca
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiKernelGenerator.py:104
↓ 1 callers
Function
coordinate_size
utils/MinkowskiEngine/src/kernel_region.hpp:174
↓ 1 callers
Method
coordinates_and_features_at
r"""Returns a coordinate and feature matrix at the specified batch index. Returns a coordinate and feature matrix at the specified `batch_ind
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiTensor.py:381
↓ 1 callers
Method
data_process
(self,data)
network/eqv_trans.py:33
↓ 1 callers
Method
data_process
(self,data)
network/eqv_trans.py:103
↓ 1 callers
Function
decomposition
()
utils/MinkowskiEngine/examples/sparse_tensor_basic.py:133
← previous
next →
201–300 of 1,370, ranked by callers