MCPcopy Create free account

hub / github.com/DylanWusee/pointconv / functions

Functions198 in github.com/DylanWusee/pointconv

MethodGroupPointGradGpuOp
tf_ops/grouping/tf_grouping.cpp:176
MethodProbSampleGpuOp
tf_ops/sampling/tf_sampling.cpp:68
MethodQueryBallPointGpuOp
tf_ops/grouping/tf_grouping.cpp:69
MethodSelectionSortGpuOp
tf_ops/grouping/tf_grouping.cpp:111
MethodThreeInterpolateGradOp
tf_ops/3d_interpolation/tf_interpolate.cpp:227
MethodThreeInterpolateOp
tf_ops/3d_interpolation/tf_interpolate.cpp:193
MethodThreeNNOp
tf_ops/3d_interpolation/tf_interpolate.cpp:159
Method__getitem__
(self, index)
scannet/scannet_dataset_rgb.py:37
Method__getitem__
(self, index)
scannet/scannet_dataset_rgb.py:98
Method__getitem__
(self, index)
scannet/scannet_dataset_sw_rgb.py:60
Method__init__
(self, root, block_points=8192, split='train', with_rgb = False)
scannet/scannet_dataset_rgb.py:14
Method__init__
(self, root, block_points=8192, split='val', with_rgb = False)
scannet/scannet_dataset_rgb.py:76
Method__init__
(self, root, split='test', num_class = 21, block_points = 8192, with_rgb = True)
scannet/scannet_dataset_sw_rgb.py:15
Method__init__
(self, mesh_vert_instances, instance_id)
scannet/visualize/util_3d.py:89
Method__len__
(self)
scannet/scannet_dataset_rgb.py:72
Method__len__
(self)
scannet/scannet_dataset_rgb.py:138
Method__len__
(self)
scannet/scannet_dataset_sw_rgb.py:150
Method__str__
(self)
scannet/visualize/util_3d.py:122
Function_gather_point_grad
(op,out_g)
tf_ops/sampling/tf_sampling.py:44
Function_group_point_grad
(op, grad_out)
tf_ops/grouping/tf_grouping.py:43
Function_three_interpolate_grad
(op, grad_out)
tf_ops/3d_interpolation/tf_interpolate.py:30
Functionangle_axis2euler
Convert angle, axis pair to Euler angles Parameters ---------- theta : scalar angle of rotation vector : 3 element sequence
scannet/eulerangles.py:382
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
utils/tf_util.py:415
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
utils/tf_util.py:465
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
utils/tf_util.py:490
Functionconv1d
1D convolution with non-linear operation. Args: inputs: 3-D tensor variable BxLxC num_output_channels: int kernel_size: int scope:
utils/tf_util.py:53
Functionconv2d
2D convolution with non-linear operation. Args: inputs: 4-D tensor variable BxHxWxC num_output_channels: int kernel_size: a list of 2
utils/tf_util.py:118
Functionconv2d_transpose
2D convolution transpose with non-linear operation. Args: inputs: 4-D tensor variable BxHxWxC num_output_channels: int kernel_size: a
utils/tf_util.py:185
Functionconv3d
3D convolution with non-linear operation. Args: inputs: 5-D tensor variable BxDxHxWxC num_output_channels: int kernel_size: a list of
utils/tf_util.py:263
Functiondropout
Dropout layer. Args: inputs: tensor is_training: boolean tf.Variable scope: string keep_prob: float in [0,1] noise_shape: list
utils/tf_util.py:612
Functioneuler2angle_axis
Return angle, axis corresponding to these Euler angles Uses the z, then y, then x convention above Parameters ---------- z : scalar
scannet/eulerangles.py:348
Functionexport_instance_ids_for_eval
(filename, label_ids, instance_ids)
scannet/visualize/util_3d.py:57
Methodfrom_json
(self, data)
scannet/visualize/util_3d.py:114
Functionfully_connected
Fully connected layer with non-linear operation. Args: inputs: 2-D tensor BxN num_outputs: int Returns: Variable tensor of size
utils/tf_util.py:322
FunctiongetDataFiles
(list_filename)
utils/provider.py:165
Functionget_batch_wdp
(dataset, idxs, start_idx, end_idx)
train_scannet_IoU.py:197
Functionget_instances
(ids, class_ids, class_labels, id2label)
scannet/visualize/util_3d.py:148
Functionget_loss
pred: BxNxC, label: BxN, smpw: BxN
models/pointconv_weight_density_n16_dp.py:49
Functionget_loss
pred: BxNxC, label: BxN, smpw: BxN
models/pointconv_weight_density_n16.py:49
Functionget_time
tf_ops/3d_interpolation/tf_interpolate.cpp:51
Functiongrouping
K: neighbor size src_xyz: original point xyz (batch_size, ndataset, 3) q_xyz: query point xyz (batch_size, npoint, 3)
utils/pointconv_util.py:113
Functionjitter_point_cloud
Randomly jitter points. jittering is per point. Input: BxNx3 array, original batch of point clouds Return: BxNx3
utils/provider.py:126
Functionkernel_density_estimation
(pts, sigma, kpoint = 32, is_norm = False)
utils/pointconv_util.py:69
Functionknn_kdtree
(nsample, xyz, new_xyz)
utils/pointconv_util.py:24
FunctionloadDataFile
(filename)
utils/provider.py:174
Functionload_ids
(filename)
scannet/visualize/util_3d.py:38
Functionmain
tf_ops/3d_interpolation/interpolate.cpp:132
Functionmain
tf_ops/grouping/test/query_ball_point.cpp:86
Functionmain
tf_ops/grouping/test/selection_sort.cpp:65
Functionmax_pool2d
2D max pooling. Args: inputs: 4-D tensor BxHxWxC kernel_size: a list of 2 ints stride: a list of 2 ints Returns: Variable ten
utils/tf_util.py:390
Functionmax_pool3d
3D max pooling. Args: inputs: 5-D tensor BxDxHxWxC kernel_size: a list of 3 ints stride: a list of 3 ints Returns: Variable t
utils/tf_util.py:440
Functionmean_var_with_update
()
utils/tf_util.py:522
Functionplaceholder_inputs
(batch_size, num_point)
models/pointconv_weight_density_n16_dp.py:12
Functionplaceholder_inputs
(batch_size, num_point)
models/pointconv_weight_density_n16.py:12
Functionpoint_cloud_label_to_surface_voxel_label
(point_cloud, label, res=0.0484)
scannet/pc_util.py:23
Functionpoint_cloud_label_to_surface_voxel_label_fast
(point_cloud, label, res=0.0484)
scannet/pc_util.py:39
Functionpoint_cloud_to_image_batch
Input is BxNx3 a batch of point cloud Output is BxIxIxnum_samplex3 Added on Feb 19
scannet/pc_util.py:155
Functionpoint_cloud_to_volume_batch
Input is BxNx3 batch of point cloud Output is Bx(vsize^3)
scannet/pc_util.py:53
Functionpoint_cloud_to_volume_v2_batch
Input is BxNx3 a batch of point cloud Output is BxVxVxVxnum_samplex3 Added on Feb 19
scannet/pc_util.py:102
Functionprint_error
(message, user_fault=False)
scannet/util.py:16
Functionprint_error
(message, user_fault=False)
scannet/visualize/util.py:16
Functionpyplot_draw_volume
vol is of size vsize*vsize*vsize output an image to output_filename
scannet/pc_util.py:397
Functionquat2euler
Return Euler angles corresponding to quaternion `q` Parameters ---------- q : 4 element sequence w, x, y, z of quaternion Re
scannet/eulerangles.py:319
Functionrandom_jitter_rgb
(batch_data, r = 2.5)
utils/provider.py:118
Functionrandom_scale_point_cloud
Randomly scale the point cloud. Scale is per point cloud. Input: BxNx3 array, original batch of point clouds Return:
utils/provider.py:152
Functionrandomf
tf_ops/3d_interpolation/tf_interpolate.cpp:48
Functionrandomf
tf_ops/grouping/test/selection_sort.cpp:9
Functionread_instance_prediction_file
(filename, pred_path)
scannet/visualize/util_3d.py:125
Functionread_label_mapping
(filename, label_from='raw_category', label_to='nyu40id')
scannet/util.py:32
Functionread_label_mapping
(filename, label_from='raw_category', label_to='nyu40id')
scannet/visualize/util.py:32
Functionread_mesh_vertices
(filename)
scannet/visualize/util_3d.py:44
Functionread_ply_rgba
read XYZRGBA point cloud from filename PLY file
scannet/pc_util.py:212
Functionread_scene_types_mapping
(filename, remove_spaces=True)
scannet/util.py:46
Functionread_scene_types_mapping
(filename, remove_spaces=True)
scannet/visualize/util.py:46
Functionreduce_sum2d_conv
(inputs, axis = None, scope = None, keepdims=False,
utils/tf_util.py:360
Functionrotate_perturbation_point_cloud
Randomly perturb the point clouds by small rotations Input: BxNx3 array, original batch of point clouds Return: B
utils/provider.py:94
Functionrotate_point_cloud
Randomly rotate the point clouds to augument the dataset rotation is per shape based along up direction Input: BxNx3 array,
utils/provider.py:33
Functionrotate_point_cloud_by_angle
Rotate the point cloud along up direction with certain angle. Input: BxNx3 array, original batch of point clouds Return:
utils/provider.py:74
Functionrotate_point_cloud_z
Randomly rotate the point clouds to augument the dataset rotation is per shape based along z direction Input: BxNx3 array,
utils/provider.py:53
Functionsampling
inputs: npoint: scalar, number of points to sample pointcloud: B * N * 3, input point cloud output: sub_pts: B * npoint * 3, sub-
utils/pointconv_util.py:101
Functionshift_point_cloud
Randomly shift point cloud. Shift is per point cloud. Input: BxNx3 array, original batch of point clouds Return:
utils/provider.py:139
Functionshuffle_data
Shuffle data and labels. Input: data: B,N,... numpy array label: B,... numpy array Return: shuffled dat
utils/provider.py:20
Functionshuffle_points
Shuffle orders of points in each point cloud -- changes FPS behavior. Use the same shuffling idx for the entire batch. Input:
utils/provider.py:8
Methodtest
(self)
tf_ops/3d_interpolation/tf_interpolate_op_test.py:6
Methodtest
(self)
tf_ops/grouping/tf_grouping_op_test.py:6
Methodtest_grad
(self)
tf_ops/3d_interpolation/tf_interpolate_op_test.py:9
Methodtest_grad
(self)
tf_ops/grouping/tf_grouping_op_test.py:9
Methodto_json
(self)
scannet/visualize/util_3d.py:102
Functiontransform_points
(matrix, points)
scannet/visualize/util_3d.py:22
Functionvisualize_instance_image
(filename, image)
scannet/util.py:70
Functionvisualize_instance_image
(filename, image)
scannet/visualize/util.py:70
Functionvisualize_label_image
(filename, image)
scannet/util.py:59
Functionvisualize_label_image
(filename, image)
scannet/visualize/util.py:59
Functionwrite_ply
input: Nx3, write points to filename as PLY format.
scannet/pc_util.py:219
Functionwrite_ply_color
Color (N,3) points with labels (N) within range 0 ~ num_classes-1 as OBJ file
scannet/pc_util.py:404
Functionwrite_ply_label
Color (N,3) points with labels (N) within range 0 ~ num_classes-1 as OBJ file
scannet/pc_util.py:233
Functionwrite_ply_label2
Color (N,3) points with labels (N) within range 0 ~ num_classes-1 as OBJ file
scannet/pc_util.py:272
Functionwrite_ply_rgb
input: Nx3, write points to filename as PLY format.
scannet/pc_util.py:226
← previous101–198 of 198, ranked by callers