MCPcopy Create free account

hub / github.com/charlesq34/pointnet2 / functions

Functions204 in github.com/charlesq34/pointnet2

↓ 1 callersFunctionrender
()
utils/show3d_balls.py:51
↓ 1 callersMethodreset
reset order of h5 files
modelnet_h5_dataset.py:63
↓ 1 callersFunctionsample_and_group
Input: npoint: int32 radius: float32 nsample: int32 xyz: (batch_size, ndataset, 3) TF tensor points: (bat
utils/pointnet_util.py:22
↓ 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
utils/pointnet_util.py:59
↓ 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_ops/grouping/tf_grouping.py:22
↓ 1 callersFunctionshuffle_data
Shuffle data and labels. Input: data: B,N,... numpy array label: B,... numpy array Return: shuffled dat
modelnet_h5_dataset.py:29
↓ 1 callersFunctiontrain
()
train.py:106
↓ 1 callersFunctiontrain
()
train_multi_gpu.py:149
↓ 1 callersFunctiontrain
()
scannet/train.py:98
↓ 1 callersFunctiontrain
()
part_seg/train.py:94
↓ 1 callersFunctiontrain
()
part_seg/train_one_hot.py:94
↓ 1 callersFunctiontrain_one_epoch
ops: dict mapping from string to tf ops
train.py:186
↓ 1 callersFunctiontrain_one_epoch
ops: dict mapping from string to tf ops
train_multi_gpu.py:260
↓ 1 callersFunctiontrain_one_epoch
ops: dict mapping from string to tf ops
scannet/train.py:211
↓ 1 callersFunctiontrain_one_epoch
ops: dict mapping from string to tf ops
part_seg/train.py:179
↓ 1 callersFunctiontrain_one_epoch
ops: dict mapping from string to tf ops
part_seg/train_one_hot.py:184
↓ 1 callersFunctionvirtual_scan
(xyz, mode=-1)
scannet/scene_util.py:20
↓ 1 callersFunctionvolume_to_point_cloud
vol is occupancy grid (value = 0 or 1) of size vsize*vsize*vsize return Nx3 numpy array.
utils/pc_util.py:56
↓ 1 callersFunctionvolume_to_point_cloud
vol is occupancy grid (value = 0 or 1) of size vsize*vsize*vsize return Nx3 numpy array.
scannet/pc_util.py:85
Method__getitem__
(self, index)
modelnet_dataset.py:93
Method__getitem__
(self, index)
scannet/scannet_dataset.py:27
Method__getitem__
(self, index)
scannet/scannet_dataset.py:83
Method__getitem__
(self, index)
scannet/scannet_dataset.py:141
Method__getitem__
(self, index)
part_seg/part_dataset.py:87
Method__getitem__
(self, index)
part_seg/part_dataset_all_normal.py:83
Method__init__
(self, list_filename, batch_size = 32, npoints = 1024, shuffle=True)
modelnet_h5_dataset.py:55
Method__init__
(self, root, batch_size = 32, npoints = 1024, split='train', normalize=True, normal_channel=False, modelnet10=
modelnet_dataset.py:24
Method__init__
(self, root, npoints=8192, split='train')
scannet/scannet_dataset.py:9
Method__init__
(self, root, npoints=8192, split='train')
scannet/scannet_dataset.py:65
Method__init__
(self, root, npoints=8192, split='train')
scannet/scannet_dataset.py:123
Method__init__
(self, root, npoints = 2500, classification = False, class_choice = None, split='train', normalize=True)
part_seg/part_dataset.py:20
Method__init__
(self, root, npoints = 2500, classification = False, split='train', normalize=True, return_cls_label = False)
part_seg/part_dataset_all_normal.py:20
Method__len__
(self)
modelnet_dataset.py:96
Method__len__
(self)
scannet/scannet_dataset.py:61
Method__len__
(self)
scannet/scannet_dataset.py:119
Method__len__
(self)
scannet/scannet_dataset.py:166
Method__len__
(self)
part_seg/part_dataset.py:112
Method__len__
(self)
part_seg/part_dataset_all_normal.py:114
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
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: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
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
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:
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
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
utils/tf_util.py:188
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:268
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:594
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:327
FunctiongetDataFiles
(list_filename)
utils/provider.py:237
Functionget_loss
pred: B*NUM_CLASSES, label: B,
models/pointnet2_cls_ssg.py:47
Functionget_loss
pred: BxNxC, label: BxN,
models/pointnet2_part_seg_msg_one_hot.py:50
Functionget_loss
pred: B*NUM_CLASSES, label: B,
models/pointnet2_cls_msg.py:42
Functionget_loss
pred: BxNxC, label: BxN, smpw: BxN
models/pointnet2_sem_seg.py:48
Functionget_loss
pred: B*NUM_CLASSES, label: B,
models/pointnet_cls_basic.py:67
Functionget_loss
pred: BxNxC, label: BxN,
models/pointnet2_part_seg.py:44
Functionjitter_point_cloud
Randomly jitter points. jittering is per point. Input: BxNx3 array, original batch of point clouds Return: BxNx3
utils/provider.py:187
FunctionloadDataFile
(filename)
modelnet_h5_dataset.py:50
FunctionloadDataFile
(filename)
utils/provider.py:246
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:366
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:417
Functionmean_var_with_update
()
utils/tf_util.py:500
Methodnum_channel
(self)
modelnet_h5_dataset.py:95
Functiononmouse
(*args)
utils/show3d_balls.py:12
Functionplaceholder_inputs
(batch_size, num_point)
models/pointnet2_cls_ssg.py:15
Functionplaceholder_inputs
(batch_size, num_point)
models/pointnet2_part_seg_msg_one_hot.py:11
Functionplaceholder_inputs
(batch_size, num_point)
models/pointnet2_cls_msg.py:11
Functionplaceholder_inputs
(batch_size, num_point)
models/pointnet2_sem_seg.py:11
Functionplaceholder_inputs
(batch_size, num_point)
models/pointnet_cls_basic.py:15
Functionplaceholder_inputs
(batch_size, num_point)
models/pointnet2_part_seg.py:11
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
utils/pc_util.py:126
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)
utils/pc_util.py:24
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
utils/pc_util.py:73
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
Functionpyplot_draw_volume
vol is of size vsize*vsize*vsize output an image to output_filename
utils/pc_util.py:292
Functionpyplot_draw_volume
vol is of size vsize*vsize*vsize output an image to output_filename
scannet/pc_util.py:347
Functionrandom_point_dropout
batch_pc: BxNx3
utils/provider.py:227
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:214
Functionread_ply_xyz
read XYZ point cloud from filename PLY file
scannet/pc_util.py:212
Functionread_ply_xyzrgb
read XYZRGB point cloud from filename PLY file
scannet/pc_util.py:224
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:162
Functionrotate_perturbation_point_cloud_with_normal
Randomly perturb the point clouds by small rotations Input: BxNx6 array, original batch of point clouds and point normals R
utils/provider.py:92
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:32
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:119
Functionrotate_point_cloud_by_angle_with_normal
Rotate the point cloud along up direction with certain angle. Input: BxNx6 array, original batch of point clouds with normal
utils/provider.py:138
Functionrotate_point_cloud_with_normal
Randomly rotate XYZ, normal point cloud. Input: batch_xyz_normal: B,N,6, first three channels are XYZ, last 3 all normal
utils/provider.py:72
Functionrotate_point_cloud_z
Randomly rotate the point clouds to augument the dataset rotation is per shape based along up direction Input: BxNx3 array,
utils/provider.py:52
Functionshift_point_cloud
Randomly shift point cloud. Shift is per point cloud. Input: BxNx3 array, original batch of point clouds Return:
utils/provider.py:200
Functionshuffle_data
Shuffle data and labels. Input: data: B,N,... numpy array label: B,... numpy array Return: shuffled dat
utils/provider.py:8
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:20
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
Functionwrite_ply
input: Nx3, write points to filename as PLY format.
utils/pc_util.py:184
← previousnext →101–200 of 204, ranked by callers