MCPcopy Create free account

hub / github.com/MVIG-SJTU/pointSIFT / functions

Functions142 in github.com/MVIG-SJTU/pointSIFT

↓ 13 callersFunctiongroup_point
Input: points: (batch_size, ndataset, channel) float32 array, points to sample from idx: (batch_size, npoint, nsample) int32 arra
tf_utils/tf_ops/grouping/tf_grouping.py:33
↓ 8 callersFunction_variable_on_cpu
Helper to create a Variable stored on CPU memory. Args: name: name of the variable shape: list of ints initializer: initializer for Vari
tf_utils/tf_util.py:10
↓ 8 callersFunctionget_time
tf_utils/tf_ops/interpolation/interpolate.cpp:12
↓ 6 callersFunctionget_time
tf_utils/tf_ops/grouping/test/query_ball_point.cpp:12
↓ 6 callersFunctionpointSIFT_module
(xyz, points, radius, out_channel, is_training, bn_decay, scope='point_sift', bn=True, use_xyz=True, use_nchw=
tf_utils/pointSIFT_util.py:71
↓ 5 callersFunction_variable_with_weight_decay
Helper to create an initialized Variable with weight decay. Note that the Variable is initialized with a truncated normal distribution. A weight
tf_utils/tf_util.py:24
↓ 4 callersMethod__len__
(self)
scannet_dataset.py:64
↓ 4 callersFunctionbatch_norm_template
Batch normalization on convolutional maps and beyond... Ref.: http://stackoverflow.com/questions/33949786/how-could-i-use-batch-normalization-in-te
tf_utils/tf_util.py:512
↓ 4 callersFunctionpointSIFT_res_module
(xyz, points, radius, out_channel, is_training, bn_decay, scope='point_sift', bn=True, use_xyz=True, same_dim=
tf_utils/pointSIFT_util.py:96
↓ 3 callersFunctionpointnet_fp_module
PointNet Feature Propogation (FP) Module Input: xyz1: (batch_size, ndataset1, 3) TF tensor xyz2: (batch_size, ndatase
tf_utils/pointSIFT_util.py:327
↓ 3 callersFunctionpointnet_sa_module
PointNet Set Abstraction (SA) Module Input: xyz: (batch_size, ndataset, 3) TF tensor points: (batch_size, ndataset, c
tf_utils/pointSIFT_util.py:211
↓ 3 callersFunctionquery_ball_point
Input: radius: float32, ball search radius nsample: int32, number of points selected in each ball region xyz1: (batch_siz
tf_utils/tf_ops/grouping/tf_grouping.py:8
↓ 3 callersFunctionrandomf
tf_utils/tf_ops/grouping/test/query_ball_point.cpp:9
↓ 3 callersFunctionrandomf
tf_utils/tf_ops/interpolation/interpolate.cpp:9
↓ 3 callersFunctionthree_interpolate
Input: points: (b,m,c) float32 array, known points idx: (b,n,3) int32 array, indices to known points weight: (b,n,3) floa
tf_utils/tf_ops/interpolation/tf_interpolate.py:19
↓ 3 callersFunctionthree_nn
Input: xyz1: (b,n,3) float32 array, unknown points xyz2: (b,m,3) float32 array, known points Output: dist: (b,n,3) fl
tf_utils/tf_ops/interpolation/tf_interpolate.py:8
↓ 2 callersFunctionbatch_norm_for_conv2d
Batch normalization on 2D convolutional maps. Args: inputs: Tensor, 4D BHWC input maps is_training: boolean tf.Varialbe, true i
tf_utils/tf_util.py:565
↓ 2 callersFunctionfarthest_point_sample
input: int32 batch_size * ndataset * 3 float32 returns: batch_size * npoint int32
tf_utils/tf_ops/sampling/tf_sampling.py:48
↓ 2 callersFunctiongather_point
input: batch_size * ndataset * 3 float32 batch_size * npoints int32 returns: batch_size * npoints * 3 float32
tf_utils/tf_ops/sampling/tf_sampling.py:29
↓ 2 callersFunctionget_deconv_dim
(dim_size, stride_size, kernel_size, padding)
tf_utils/tf_util.py:236
↓ 2 callersFunctionget_time
tf_utils/tf_ops/grouping/test/selection_sort.cpp:12
↓ 2 callersFunctionpointSIFT_group
(radius, xyz, points, use_xyz=True)
tf_utils/pointSIFT_util.py:15
↓ 1 callersMethodave_gradient
(tower_grad)
train_and_eval_scannet.py:145
↓ 1 callersFunctionbatch_norm_for_conv1d
Batch normalization on 1D convolutional maps. Args: inputs: Tensor, 3D BLC input maps is_training: boolean tf.Varialbe, true in
tf_utils/tf_util.py:548
↓ 1 callersFunctionbatch_norm_for_conv3d
Batch normalization on 3D convolutional maps. Args: inputs: Tensor, 5D BDHWC input maps is_training: boolean tf.Varialbe, true
tf_utils/tf_util.py:580
↓ 1 callersFunctionbatch_norm_for_fc
Batch normalization on FC data. Args: inputs: Tensor, 2D BxC input is_training: boolean tf.Varialbe, true indicates training ph
tf_utils/tf_util.py:534
↓ 1 callersMethodbuild_g_cpu
(self)
train_and_eval_scannet.py:159
↓ 1 callersMethodbuild_g_gpu
(self, gpu_idx)
train_and_eval_scannet.py:170
↓ 1 callersMethodbuild_graph
(self)
train_and_eval_scannet.py:189
↓ 1 callersMethodevaluate_one_epoch
(self, sess, test_writer, step, epoch)
train_and_eval_scannet.py:261
↓ 1 callersFunctionfun
(xyz1,xyz2,pts2)
tf_utils/tf_ops/interpolation/visu_interpolation.py:16
↓ 1 callersMethodget_batch_wdp
(self, dataset, idxs, start_idx, end_idx)
train_and_eval_scannet.py:113
↓ 1 callersMethodget_bn_decay
(self)
train_and_eval_scannet.py:103
↓ 1 callersMethodget_learning_rate
(self)
train_and_eval_scannet.py:93
↓ 1 callersFunctionget_weights_cpu
CONSTANT WEIGHT TODO input: dist (b,n,3) output: weight (b,n,3)
tf_utils/tf_ops/interpolation/interpolate.cpp:69
↓ 1 callersFunctiongroup_point_cpu
input: points (b,n,c), idx (b,m,nsample) output: out (b,m,nsample,c)
tf_utils/tf_ops/grouping/test/query_ball_point.cpp:52
↓ 1 callersFunctiongroup_point_grad_cpu
input: grad_out (b,m,nsample,c), idx (b,m,nsample), output: grad_points (b,n,c)
tf_utils/tf_ops/grouping/test/query_ball_point.cpp:70
↓ 1 callersFunctioninterpolate_cpu
input: points (b,m,c), idx (b,n,3), weight (b,n,3) output: out (b,n,c)
tf_utils/tf_ops/interpolation/interpolate.cpp:84
↓ 1 callersFunctioninterpolate_grad_cpu
input: grad_out (b,n,c), idx (b,n,3), weight (b,n,3) output: grad_points (b,m,c)
tf_utils/tf_ops/interpolation/interpolate.cpp:108
↓ 1 callersFunctionknn_point
Input: k: int32, number of k in k-nn search xyz1: (batch_size, ndataset, c) float32 array, input points xyz2: (batch_size
tf_utils/tf_ops/grouping/tf_grouping.py:49
↓ 1 callersMethodload_data
(self)
train_and_eval_scannet.py:85
↓ 1 callersFunctionload_h5
(h5_filename)
tf_utils/provider.py:262
↓ 1 callersFunctionpointSIFT_group_with_idx
(xyz, idx, points, use_xyz=True)
tf_utils/pointSIFT_util.py:30
↓ 1 callersFunctionpointSIFT_select
:param xyz: (b, n, 3) float :param radius: float :return: (b, n, 8) int
tf_utils/tf_ops/pointSIFT_op/pointSIFT_op.py:16
↓ 1 callersFunctionpointSIFT_select_four
:param xyz: (b, n, 3) float :param radius: float :return: idx: (b, n, 32) int
tf_utils/tf_ops/pointSIFT_op/pointSIFT_op.py:40
↓ 1 callersFunctionquery_ball_point_cpu
input: radius (1), nsample (1), xyz1 (b,n,3), xyz2 (b,m,3) output: idx (b,m,nsample)
tf_utils/tf_ops/grouping/test/query_ball_point.cpp:19
↓ 1 callersFunctionsample_and_group
Input: npoint: int32 radius: float32 nsample: int32 xyz: (batch_size, ndataset, 3) TF tensor points: (bat
tf_utils/pointSIFT_util.py:145
↓ 1 callersFunctionsample_and_group_all
Inputs: xyz: (batch_size, ndataset, 3) TF tensor points: (batch_size, ndataset, channel) TF tensor, if None will just use xyz as
tf_utils/pointSIFT_util.py:182
↓ 1 callersFunctionselect_top_k
Input: k: int32, number of k SMALLEST elements selected dist: (b,m,n) float32 array, distance matrix, m query points, n dataset p
tf_utils/tf_ops/grouping/tf_grouping.py:22
↓ 1 callersFunctionselection_sort_cpu
input: k (1), distance matrix dist (b,m,n) output: idx (b,m,n), val (b,m,n)
tf_utils/tf_ops/grouping/test/selection_sort.cpp:20
↓ 1 callersFunctionthreeinterpolate_cpu
input: points (b,m,c), idx (b,n,3), weight (b,n,3) output: out (b,n,c)
tf_utils/tf_ops/interpolation/tf_interpolate.cpp:107
↓ 1 callersFunctionthreeinterpolate_grad_cpu
input: grad_out (b,n,c), idx (b,n,3), weight (b,n,3) output: grad_points (b,m,c)
tf_utils/tf_ops/interpolation/tf_interpolate.cpp:131
↓ 1 callersFunctionthreenn_cpu
Find three nearest neigbors with square distance input: xyz1 (b,n,3), xyz2(b,m,3) output: dist (b,n,3), idx (b,n,3)
tf_utils/tf_ops/interpolation/interpolate.cpp:21
↓ 1 callersFunctionthreenn_cpu
Find three nearest neigbors with square distance input: xyz1 (b,n,3), xyz2(b,m,3) output: dist (b,n,3), idx (b,n,3)
tf_utils/tf_ops/interpolation/tf_interpolate.cpp:60
↓ 1 callersMethodtraining
(self)
train_and_eval_scannet.py:212
MethodCompute
tf_utils/tf_ops/sampling/tf_sampling.cpp:69
MethodCompute
tf_utils/tf_ops/sampling/tf_sampling.cpp:101
MethodCompute
tf_utils/tf_ops/sampling/tf_sampling.cpp:129
MethodCompute
tf_utils/tf_ops/sampling/tf_sampling.cpp:154
MethodCompute
tf_utils/tf_ops/grouping/tf_grouping.cpp:77
MethodCompute
tf_utils/tf_ops/grouping/tf_grouping.cpp:116
MethodCompute
tf_utils/tf_ops/grouping/tf_grouping.cpp:147
MethodCompute
tf_utils/tf_ops/grouping/tf_grouping.cpp:178
MethodCompute
tf_utils/tf_ops/interpolation/tf_interpolate.cpp:161
MethodCompute
tf_utils/tf_ops/interpolation/tf_interpolate.cpp:195
MethodCompute
tf_utils/tf_ops/interpolation/tf_interpolate.cpp:229
MethodCompute
tf_utils/tf_ops/pointSIFT_op/main.cpp:71
MethodCompute
tf_utils/tf_ops/pointSIFT_op/main.cpp:106
MethodCompute
tf_utils/tf_ops/pointSIFT_op/main.cpp:132
MethodCompute
tf_utils/tf_ops/pointSIFT_op/main.cpp:158
MethodCubeSelectFourOp
tf_utils/tf_ops/pointSIFT_op/main.cpp:154
MethodCubeSelectOp
tf_utils/tf_ops/pointSIFT_op/main.cpp:102
MethodCubeSelectTwoOp
tf_utils/tf_ops/pointSIFT_op/main.cpp:128
MethodFarthestPointSampleGpuOp
tf_utils/tf_ops/sampling/tf_sampling.cpp:97
MethodGatherPointGpuOp
tf_utils/tf_ops/sampling/tf_sampling.cpp:128
MethodGatherPointGradGpuOp
tf_utils/tf_ops/sampling/tf_sampling.cpp:153
MethodGroupPointGpuOp
tf_utils/tf_ops/grouping/tf_grouping.cpp:145
MethodGroupPointGradGpuOp
tf_utils/tf_ops/grouping/tf_grouping.cpp:176
MethodProbSampleGpuOp
tf_utils/tf_ops/sampling/tf_sampling.cpp:68
MethodQueryBallPointGpuOp
tf_utils/tf_ops/grouping/tf_grouping.cpp:69
MethodSelectionKRadiusOp
tf_utils/tf_ops/pointSIFT_op/main.cpp:68
MethodSelectionSortGpuOp
tf_utils/tf_ops/grouping/tf_grouping.cpp:111
MethodThreeInterpolateGradOp
tf_utils/tf_ops/interpolation/tf_interpolate.cpp:227
MethodThreeInterpolateOp
tf_utils/tf_ops/interpolation/tf_interpolate.cpp:193
MethodThreeNNOp
tf_utils/tf_ops/interpolation/tf_interpolate.cpp:159
Method__getitem__
(self, index)
scannet_dataset.py:26
Method__getitem__
(self, index)
scannet_dataset.py:88
Method__init__
(self)
train_and_eval_scannet.py:50
Method__init__
(self, root, npoints=8192, split='train')
scannet_dataset.py:7
Method__init__
(self, root, npoints=8192, split='train')
scannet_dataset.py:69
Method__len__
(self)
scannet_dataset.py:129
Function_gather_point_grad
(op,out_g)
tf_utils/tf_ops/sampling/tf_sampling.py:44
Function_group_point_grad
(op, grad_out)
tf_utils/tf_ops/grouping/tf_grouping.py:44
Function_three_interpolate_grad
(op, grad_out)
tf_utils/tf_ops/interpolation/tf_interpolate.py:30
Functionavg_pool2d
2D avg pooling. Args: inputs: 4-D tensor BxHxWxC kernel_size: a list of 2 ints stride: a list of 2 ints Returns: Variable ten
tf_utils/tf_util.py:391
Functionavg_pool3d
3D avg pooling. Args: inputs: 5-D tensor BxDxHxWxC kernel_size: a list of 3 ints stride: a list of 3 ints Returns: Variable t
tf_utils/tf_util.py:442
Functionbatch_norm_template_unused
NOTE: this is older version of the util func. it is deprecated. Batch normalization on convolutional maps and beyond... Ref.: http://stackoverflo
tf_utils/tf_util.py:468
Functionconv1d
1D convolution with non-linear operation. Args: inputs: 3-D tensor variable BxLxC num_output_channels: int kernel_size: int scope:
tf_utils/tf_util.py:52
Functionconv2d
2D convolution with non-linear operation. Args: inputs: 4-D tensor variable BxHxWxC num_output_channels: int kernel_size: a list of 2
tf_utils/tf_util.py:120
Functionconv2d_transpose
2D convolution transpose with non-linear operation. Args: inputs: 4-D tensor variable BxHxWxC num_output_channels: int kernel_size: a
tf_utils/tf_util.py:188
next →1–100 of 142, ranked by callers