MCPcopy Create free account

hub / github.com/HpWang-whu/RoReg / functions

Functions1,370 in github.com/HpWang-whu/RoReg

↓ 2 callersFunctionload_model
Creates and returns an instance of the model given its class name.
backbone/fcgf/__init__.py:16
↓ 2 callersFunctionmake_data_loader
(phase, augment_data, batch_size, shuffle, num_workers, repeat, config)
utils/MinkowskiEngine/examples/vae.py:156
↓ 2 callersMethodnetwork_initialization
(self, in_channels, out_channels, D)
utils/MinkowskiEngine/examples/resnet.py:67
↓ 2 callersMethodnodes
utils/MinkowskiEngine/src/3rdparty/robin_hood.h:1135
↓ 2 callersFunctionnot_reentrant_error
(error_str='')
utils/MinkowskiEngine/MinkowskiEngine/utils/gradcheck.py:196
↓ 2 callersFunctionones
utils/MinkowskiEngine/src/coordinate_map_manager.cpp:45
↓ 2 callersMethodorigin
(self)
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiCoordinateManager.py:270
↓ 2 callersMethodorigin
utils/MinkowskiEngine/src/coordinate_map_manager.cpp:470
↓ 2 callersMethodransac
(self,dataset,keynum,max_iter=1000)
test/estimator.py:245
↓ 2 callersFunctionread
(*parts)
utils/MinkowskiEngine/setup.py:76
↓ 2 callersMethodrecall
(self,data)
train/val.py:13
↓ 2 callersFunctionresample_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 callersMethodreset_permutation
(self)
utils/MinkowskiEngine/examples/reconstruction.py:63
↓ 2 callersMethodreset_permutation
(self)
utils/MinkowskiEngine/examples/common.py:76
↓ 2 callersFunctionrun_command
(*args)
utils/MinkowskiEngine/setup.py:89
↓ 2 callersFunctionsave_ctx
( ctx, # function object context kernel_generator: KernelGenerator, in_coords_key: CoordinateMapK
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiKernelGenerator.py:383
↓ 2 callersFunctionscore_mat
(source,target)
network/rot_coh_match.py:8
↓ 2 callersFunctionset_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 callersFunctionset_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 callersMethodsize
utils/MinkowskiEngine/src/3rdparty/robin_hood.h:1878
↓ 2 callersMethodsize
Returns the size of the array.
utils/MinkowskiEngine/src/primitives/small_vector.hpp:193
↓ 2 callersMethodslice
r""" Args: :attr:`X` (:attr:`MinkowskiEngine.SparseTensor`): a sparse tensor that discretized the original input.
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiSparseTensor.py:530
↓ 2 callersFunctionsparse_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 callersFunctionsparse_tensor_operation_mode
r"""Return the current sparse tensor operation mode.
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiTensor.py:103
↓ 2 callersMethodstride_map
(self, in_key: CoordinateMapKey, stride_key: CoordinateMapKey)
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiCoordinateManager.py:429
↓ 2 callersFunctionswap
utils/MinkowskiEngine/src/primitives/small_vector.hpp:331
↓ 2 callersFunctiontest
(net, device, config, phase="val")
utils/MinkowskiEngine/examples/classification_modelnet40.py:238
↓ 2 callersMethodtic
(self)
utils/MinkowskiEngine/examples/common.py:46
↓ 2 callersFunctionto_cuda
(data)
utils/utils.py:104
↓ 2 callersMethodtoc
(self, average=False)
utils/MinkowskiEngine/examples/common.py:51
↓ 2 callersMethodunion_map
(self, in_keys: list, out_key)
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiCoordinateManager.py:432
↓ 2 callersFunctionwarn_bc_breaking
()
utils/MinkowskiEngine/MinkowskiEngine/utils/gradcheck.py:239
↓ 1 callersMethodBatch_Des2R_torch
(self,des1_eqv,des2_eqv)
test/estimator.py:85
↓ 1 callersMethodClean_cache
(self)
train/trainset/GF_ET.py:291
↓ 1 callersMethodDR_statictic
(self,DR_indexs)
test/estimator.py:119
↓ 1 callersMethodDeltaR
(self,R,index)
dataops/dataloader.py:195
↓ 1 callersMethodDeltaR
(self,R,index)
train/trainset/GF_ET.py:156
↓ 1 callersMethodDes2DR
(self,Des1,Des2)
network/group_feat.py:55
↓ 1 callersMethodFCGF_Group_Feature_Extractor
(self,Point,Keys_index)
train/trainset/GF_ET.py:70
↓ 1 callersMethodFeature_extracting
(self, data_loader)
testset.py:124
↓ 1 callersMethodGen_trainset
(self)
train/trainset/GF_ET.py:163
↓ 1 callersMethodGen_valset
(self)
train/trainset/GF_ET.py:250
↓ 1 callersMethodPartII_SO3_Conv
(self,data)
network/eqv_trans.py:39
↓ 1 callersMethodPartII_SO3_Conv
(self,data)
network/eqv_trans.py:111
↓ 1 callersMethodR2DR_id
(self,R)
dataops/dataloader.py:147
↓ 1 callersMethodR2DR_id
(self,R)
train/trainset/RM.py:162
↓ 1 callersMethodR_cal
(self,key_m0,key_m1,center0,center1,scores)
test/estimator.py:45
↓ 1 callersMethodRargu_on1
(self,keys1,feats1)
train/trainset/RM.py:154
↓ 1 callersMethodRt_cal
(self,key_m0,key_m1,scores)
test/estimator.py:53
↓ 1 callersMethodSO3_Conv
(self,data)
network/group_feat.py:26
↓ 1 callersMethodSVDR_w
(self,beforerot,afterrot,scores)
test/estimator.py:39
↓ 1 callersMethodTV_RM
(self,batchnum=30000)
train/trainset/RM.py:90
↓ 1 callersFunctionThreepps2Tran
(kps0_init,kps1_init)
utils/utils.py:242
↓ 1 callersMethodThreepps2Tran
(self,kps0_init,kps1_init)
test/estimator.py:139
↓ 1 callersMethod__init__
(self, log_dir)
utils/utils.py:201
↓ 1 callersMethod__init__
(self, in_channel, out_channel)
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiNonlinearity.py:176
↓ 1 callersMethod__init__
(self, inplanes, planes, stride=1, dilatio
utils/MinkowskiEngine/MinkowskiEngine/modules/resnet_block.py:32
↓ 1 callersMethod__init__
(self, in_channels, out_channels, D=3)
utils/MinkowskiEngine/examples/multigpu_lightning.py:79
↓ 1 callersMethod__init__
(self, data_source, shuffle=False)
utils/MinkowskiEngine/examples/reconstruction.py:58
↓ 1 callersMethod__init__
(self, cfg)
checkpoints/FCGF/RD_former/rot_detect.py:7
↓ 1 callersMethod__init__
(self,in_dim,out_dim)
network/ops.py:12
↓ 1 callersMethod__init__
(self, cfg)
network/rot_detect.py:9
↓ 1 callersMethod__init__
(self, cfg)
network/eqv_trans.py:79
↓ 1 callersFunction_calculate_correct_fan
(tensor, mode)
utils/MinkowskiEngine/MinkowskiEngine/utils/init.py:25
↓ 1 callersFunction_calculate_fan_in_and_fan_out
(tensor)
utils/MinkowskiEngine/MinkowskiEngine/utils/init.py:5
↓ 1 callersMethod_get_coordinate_field
(self)
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiTensorField.py:251
↓ 1 callersMethod_get_coordinates
(self)
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiTensor.py:225
↓ 1 callersMethod_init_dataset
(self)
train/trainer.py:53
↓ 1 callersMethod_init_logger
(self)
train/trainer.py:86
↓ 1 callersMethod_init_network
(self)
train/trainer.py:61
↓ 1 callersMethod_is_same_key
(self, other)
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiTensorField.py:370
↓ 1 callersMethod_load_model
(self)
test/extractor.py:22
↓ 1 callersMethod_load_model
(self)
test/detector.py:19
↓ 1 callersMethod_load_model
(self)
test/matcher.py:121
↓ 1 callersMethod_load_model
(self)
test/estimator.py:284
↓ 1 callersMethod_load_model
(self)
train/trainer.py:68
↓ 1 callersMethod_sample_noise
(self, num, noise_params)
utils/MinkowskiEngine/examples/training.py:109
↓ 1 callersMethod_sample_xs
Return random numbers between line_params[0], line_params[1]
utils/MinkowskiEngine/examples/training.py:113
↓ 1 callersFunction_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 callersMethod_uniform_to_angle
(self, u)
utils/MinkowskiEngine/examples/training.py:105
↓ 1 callersMethodallocate
clang-format off
utils/MinkowskiEngine/src/coordinate_map.hpp:184
↓ 1 callersMethodallocate
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 callersFunctionattention
(query, key, value)
network/rot_coh_match.py:84
↓ 1 callersMethodbatch_create
(self,feats0_fcgf,feats1_fcgf,feats0_yomo,feats1_yomo,index_pre,start,end)
test/estimator.py:293
↓ 1 callersMethodbatch_feature_extraction
(self)
testset.py:184
↓ 1 callersMethodbegin
utils/MinkowskiEngine/src/3rdparty/robin_hood.h:1756
↓ 1 callersMethodcalcNumBytesInfo
utils/MinkowskiEngine/src/3rdparty/robin_hood.h:1923
↓ 1 callersMethodcalcNumBytesTotal
utils/MinkowskiEngine/src/3rdparty/robin_hood.h:1936
↓ 1 callersFunctioncat
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 callersMethodcend
utils/MinkowskiEngine/src/3rdparty/robin_hood.h:1806
↓ 1 callersMethodcenter_cal
(self,key_m0,key_m1,scores)
test/estimator.py:32
↓ 1 callersFunctioncheck_undefined_grad_support
(output_to_check)
utils/MinkowskiEngine/MinkowskiEngine/utils/gradcheck.py:246
↓ 1 callersFunctionclear_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 callersFunctionconv_on_coords
()
utils/MinkowskiEngine/examples/convolution.py:93
↓ 1 callersFunctionconvert_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 callersFunctioncoordinate_size
utils/MinkowskiEngine/src/kernel_region.hpp:174
↓ 1 callersMethodcoordinates_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 callersMethoddata_process
(self,data)
network/eqv_trans.py:33
↓ 1 callersMethoddata_process
(self,data)
network/eqv_trans.py:103
↓ 1 callersFunctiondecomposition
()
utils/MinkowskiEngine/examples/sparse_tensor_basic.py:133
← previousnext →201–300 of 1,370, ranked by callers