MCPcopy Create free account

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

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

↓ 316 callersMethodsize
utils/MinkowskiEngine/src/types.hpp:68
↓ 201 callersMethodappend
utils/MinkowskiEngine/src/utils.hpp:114
↓ 89 callersMethoddouble
(self)
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiTensor.py:449
↓ 77 callersMethodget_key
utils/MinkowskiEngine/src/coordinate_map_key.hpp:100
↓ 66 callersMethodend
utils/MinkowskiEngine/src/utils.hpp:259
↓ 65 callersMethodrequires_grad_
(self, requires_grad: bool = True)
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiTensor.py:442
↓ 57 callersMethoddata
utils/MinkowskiEngine/src/types.hpp:69
↓ 55 callersFunctiondata_loader
( nchannel=3, max_label=5, is_classification=True, seed=-1, batch_size=2, dtype=torch.
utils/MinkowskiEngine/tests/python/common.py:57
↓ 54 callersMethodexists
utils/MinkowskiEngine/src/coordinate_map_manager.hpp:282
↓ 45 callersFunctiongradcheck
r"""Check gradients computed via small finite differences against analytical gradients w.r.t. tensors in :attr:`inputs` that are of floating point
utils/MinkowskiEngine/MinkowskiEngine/utils/gradcheck.py:38
↓ 44 callersMethodbegin
utils/MinkowskiEngine/src/utils.hpp:258
↓ 41 callersMethodbackward
(ctx, grad_out_feat)
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiUnion.py:63
↓ 39 callersMethodfind
utils/MinkowskiEngine/src/3rdparty/robin_hood.h:1730
↓ 35 callersMethoddevice
(self)
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiTensor.py:464
↓ 34 callersMethodsize
(self)
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiTensor.py:456
↓ 32 callersFunction_wrap_tensor
(input, F)
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiFunctional.py:30
↓ 32 callersFunctionconv
()
utils/MinkowskiEngine/examples/convolution.py:48
↓ 32 callersFunctionget_norm
(norm_type, num_feats, bn_momentum=0.05, D=-1)
backbone/fcgf/common.py:3
↓ 28 callersMethodget_tensor_stride
utils/MinkowskiEngine/src/coordinate_map_key.hpp:135
↓ 27 callersMethoddtype
(self)
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiTensor.py:468
↓ 26 callersMethoddetach
(self)
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiTensor.py:471
↓ 25 callersFunctionbatched_coordinates
r"""Create a `ME.SparseTensor` coordinates from a sequence of coordinates Given a list of either numpy or pytorch tensor coordinates, return the
utils/MinkowskiEngine/MinkowskiEngine/utils/collation.py:30
↓ 25 callersMethodpush_back
Adds a value to the back.
utils/MinkowskiEngine/src/primitives/small_vector.hpp:248
↓ 23 callersMethodsize
utils/MinkowskiEngine/src/coordinate_map_manager.hpp:331
↓ 22 callersFunctionzeros
utils/MinkowskiEngine/src/coordinate_map_manager.cpp:43
↓ 21 callersMethodcoordinate_size
utils/MinkowskiEngine/src/coordinate_map.hpp:178
↓ 21 callersMethodfloat
(self)
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiTensor.py:445
↓ 21 callersMethodrequires_grad
(self)
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiTensor.py:439
↓ 20 callersFunctionconvert_to_int_list
( arg: Union[int, Sequence, np.ndarray, torch.Tensor], dimension: int )
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiCommon.py:39
↓ 20 callersFunctionget_minkowski_function
(name, variable)
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiCommon.py:110
↓ 20 callersFunctionmake_non_exists_dir
(fn)
utils/utils.py:9
↓ 18 callersMethodget_kps
(self)
testset.py:71
↓ 18 callersMethodkernel_map
* Kernel map ****************************************************************************/
utils/MinkowskiEngine/src/coordinate_map_cpu.hpp:571
↓ 16 callersFunctionfail_test
(msg)
utils/MinkowskiEngine/MinkowskiEngine/utils/gradcheck.py:90
↓ 16 callersMethodget_coordinate_size
coordinate_size functions
utils/MinkowskiEngine/src/coordinate_map_key.hpp:80
↓ 16 callersMethodinsert_and_map
utils/MinkowskiEngine/src/coordinate_map_cpu.hpp:356
↓ 16 callersMethodmax
utils/MinkowskiEngine/src/3rdparty/cudf/utilities/legacy/wrapper_types.hpp:502
↓ 14 callersMethodinsert_and_map
r"""create a new coordinate map and returns (key, (map, inverse_map)). :attr:`coordinates`: `torch.Tensor` (Int tensor. `CUDA` if coo
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiCoordinateManager.py:153
↓ 14 callersMethodsize
utils/MinkowskiEngine/src/coordinate_map_cpu.hpp:877
↓ 13 callersFunctionload_file
(file_name)
utils/MinkowskiEngine/tests/python/common.py:36
↓ 13 callersMethodrun
(self,dataset,keynum=5000)
test/matcher.py:49
↓ 12 callersMethod_make_layer
(self, block, planes, blocks, stride=1, dilation=1, bn_momentum=0.1)
utils/MinkowskiEngine/examples/resnet.py:114
↓ 12 callersMethodget_coordinates
(self, coords_key_or_tensor_strides)
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiCoordinateManager.py:326
↓ 12 callersMethodinsert
utils/MinkowskiEngine/src/3rdparty/robin_hood.h:1661
↓ 12 callersMethodmin
utils/MinkowskiEngine/src/3rdparty/cudf/utilities/legacy/wrapper_types.hpp:510
↓ 11 callersMethodCoordinateMapKey
utils/MinkowskiEngine/src/coordinate_map_key.hpp:54
↓ 11 callersMethodCoordinateMapManager
allocator backend will be ignored when coordinate map backend is CPU
utils/MinkowskiEngine/src/coordinate_map_manager.hpp:130
↓ 11 callersFunctionread_pickle
(pkl_path)
utils/utils.py:22
↓ 10 callersMethodfind
utils/MinkowskiEngine/src/coordinate_map_manager.hpp:273
↓ 10 callersMethodis_key_set
utils/MinkowskiEngine/src/coordinate_map_key.hpp:107
↓ 10 callersFunctionload_file
(file_name)
utils/MinkowskiEngine/tests/cpp/utils.py:41
↓ 10 callersMethodresize
\brief Resizes the array, making it either larger or smaller. Initializes new elements with `newval`.
utils/MinkowskiEngine/src/primitives/small_vector.hpp:136
↓ 10 callersMethodset_key
key functions
utils/MinkowskiEngine/src/coordinate_map_key.hpp:83
↓ 10 callersMethodsparse
r"""Converts the current sparse tensor field to a sparse tensor.
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiTensorField.py:254
↓ 10 callersFunctionsparse_quantize
r"""Given coordinates, and features (optionally labels), the function generates quantized (voxelized) coordinates. Args: :attr:`coord
utils/MinkowskiEngine/MinkowskiEngine/utils/quantization.py:125
↓ 9 callersFunctioncompute_R_diff
(R_gt, R)
utils/r_eval.py:108
↓ 8 callersMethod__init__
r"""a high-dimensional sparse average pooling layer. Args: :attr:`kernel_size` (int, optional): the size of the kernel in the
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiPooling.py:232
↓ 8 callersMethod__init__
(self,cfg)
network/rot_coh_match.py:324
↓ 8 callersFunction_argparse
(pattern, argv, is_flag=True, is_list=False)
utils/MinkowskiEngine/setup.py:93
↓ 8 callersMethodget_transform
(self,id0,id1)
dataops/dataset.py:26
↓ 8 callersMethodtic
utils/MinkowskiEngine/src/utils.hpp:43
↓ 8 callersMethodtoc
utils/MinkowskiEngine/src/utils.hpp:45
↓ 7 callersFunctionArrToString
utils/MinkowskiEngine/src/utils.hpp:86
↓ 7 callersMethodKNN
(self,k)
utils/knn_search.py:169
↓ 7 callersMethodcapacity
utils/MinkowskiEngine/src/3rdparty/robin_hood.h:1909
↓ 7 callersMethodfind
utils/MinkowskiEngine/src/coordinate_map_cpu.hpp:391
↓ 7 callersMethodgetFirst
utils/MinkowskiEngine/src/3rdparty/robin_hood.h:952
↓ 7 callersFunctionget_block
(norm_type, inplanes, planes, stride=1, dilation=1,
backbone/fcgf/residual_block.py:64
↓ 7 callersMethodget_conv_block
(self, in_channel, out_channel, kernel_size, stride)
utils/MinkowskiEngine/examples/classification_modelnet40.py:96
↓ 7 callersFunctionget_dataset_name
(dataset_name,origin_data_dir)
dataops/dataset.py:132
↓ 7 callersMethodinsert
* @brief given a key iterator begin-end pair and a value iterator begin-end * pair, insert all elements. * * @return none */
utils/MinkowskiEngine/src/coordinate_map_cpu.hpp:330
↓ 7 callersMethodorigin_map
utils/MinkowskiEngine/src/coordinate_map_cpu.hpp:726
↓ 7 callersFunctionspmm
( rows: torch.Tensor, cols: torch.Tensor, vals: torch.Tensor, size: torch.Size, mat: torch
utils/MinkowskiEngine/MinkowskiEngine/sparse_matrix_functions.py:32
↓ 7 callersMethodstr
utils/MinkowskiEngine/src/utils.hpp:123
↓ 7 callersMethodstride
r"""Generate a new coordinate map and returns the key. :attr:`coordinate_map_key` (:attr:`MinkowskiEngine.CoordinateMapKey`): input m
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiCoordinateManager.py:254
↓ 7 callersMethodstride
utils/MinkowskiEngine/src/coordinate_map_manager.cpp:404
↓ 7 callersFunctionswap
utils/MinkowskiEngine/src/3rdparty/robin_hood.h:654
↓ 7 callersFunctionto_sparse_coo
(data)
utils/MinkowskiEngine/examples/sparse_tensor_basic.py:40
↓ 7 callersFunctiontransform_points
(pts,transform)
utils/utils.py:38
↓ 6 callersFunctionadd_argument_group
(name)
parses/parses_train_rm.py:11
↓ 6 callersFunctionadd_argument_group
(name)
parses/parses_test.py:10
↓ 6 callersFunctioncriterion
Calculate cross entropy loss, apply label smoothing if needed.
utils/MinkowskiEngine/examples/classification_modelnet40.py:257
↓ 6 callersFunctionend
utils/MinkowskiEngine/src/kernel_region.hpp:165
↓ 6 callersMethodget_target
(self, out, target_key, kernel_size=1)
utils/MinkowskiEngine/examples/vae.py:462
↓ 6 callersMethodget_target
(self, out, target_key, kernel_size=1)
utils/MinkowskiEngine/examples/reconstruction.py:430
↓ 6 callersMethodget_target
(self, out, target_key, kernel_size=1)
utils/MinkowskiEngine/examples/completion.py:357
↓ 6 callersMethodpdist
(self, A, B, dist_type='L2')
utils/knn_search.py:17
↓ 6 callersFunctionquaternion_from_matrix
Return quaternion from rotation matrix. If isprecise is True, the input matrix is assumed to be a precise rotation matrix and a faster algori
utils/r_eval.py:5
↓ 6 callersFunctionsave_pickle
(data, pkl_path)
utils/utils.py:26
↓ 6 callersMethodslice
(self)
utils/MinkowskiEngine/tests/python/tensor_field.py:149
↓ 5 callersMethod__init__
(self,keys: list or tuple)
train/loss.py:12
↓ 5 callersMethoddata
helper functions
utils/MinkowskiEngine/src/coordinate.hpp:51
↓ 5 callersMethodempty
utils/MinkowskiEngine/src/3rdparty/robin_hood.h:1888
↓ 5 callersMethodget
* @brief Get raw pointer to underlying NVTX attributes object. * */
utils/MinkowskiEngine/src/3rdparty/cudf/detail/nvtx/nvtx3.hpp:1634
↓ 5 callersMethodkernel_map
r"""Get kernel in-out maps for the specified coords keys or tensor strides. returns dict{kernel_index: in_out_tensor} where in_out_tensor[0]
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiCoordinateManager.py:377
↓ 5 callersMethodlog
(self,data, prefix='train',step=None,verbose=False)
utils/utils.py:206
↓ 5 callersMethodnext
forwards the index by one, wrapping around at the end
utils/MinkowskiEngine/src/3rdparty/robin_hood.h:1301
↓ 5 callersFunctionnorm
(a,axis=-1)
utils/utils.py:252
↓ 5 callersMethodnum_steps
utils/MinkowskiEngine/src/3rdparty/robin_hood.h:1243
↓ 5 callersMethodorigin_map
(self, key: CoordinateMapKey)
utils/MinkowskiEngine/MinkowskiEngine/MinkowskiCoordinateManager.py:423
next →1–100 of 1,370, ranked by callers