Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/WangYueFt/dgcnn
/ functions
Functions
215 in github.com/WangYueFt/dgcnn
⨍
Functions
215
◇
Types & classes
9
↓ 75 callers
Method
write
Write PLY data to a writeable file-like object or filename.
tensorflow/utils/plyfile.py:291
↓ 27 callers
Method
close
(self)
pytorch/util.py:45
↓ 11 callers
Function
printout
(flog, data)
tensorflow/part_seg/train_multi_gpu.py:84
↓ 10 callers
Function
_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
tensorflow/utils/tf_util.py:12
↓ 9 callers
Method
dtype
Return the numpy dtype of the in-memory representation of the data. (If there are no list properties, and the PLY format is
tensorflow/utils/plyfile.py:446
↓ 9 callers
Function
log_string
(out_str)
tensorflow/train.py:67
↓ 8 callers
Function
_lookup_type
(type_str)
tensorflow/utils/plyfile.py:80
↓ 8 callers
Function
printout
(flog, data)
tensorflow/part_seg/test.py:56
↓ 6 callers
Method
cprint
(self, text)
pytorch/util.py:40
↓ 6 callers
Function
log_string
(out_str)
tensorflow/evaluate.py:51
↓ 5 callers
Function
_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
tensorflow/utils/tf_util.py:26
↓ 5 callers
Function
log_string
(out_str)
tensorflow/sem_seg/train.py:95
↓ 4 callers
Function
batch_norm_dist_template
The batch normalization for distributed training. Args: inputs: Tensor, k-D input ... x C could be BC or BHWC or BDHWC is_traini
tensorflow/utils/tf_util.py:502
↓ 4 callers
Function
batch_norm_template
Batch normalization on convolutional maps and beyond... Ref.: http://stackoverflow.com/questions/33949786/how-could-i-use-batch-normalization-in-te
tensorflow/utils/tf_util.py:462
↓ 4 callers
Method
dtype
Return the numpy dtype description for this property (as a tuple of strings).
tensorflow/utils/plyfile.py:770
↓ 4 callers
Function
get_graph_feature
(x, k=20, idx=None)
pytorch/model.py:30
↓ 4 callers
Method
list_dtype
Return the pair (len_dtype, val_dtype) (both numpy-friendly strings).
tensorflow/utils/plyfile.py:847
↓ 4 callers
Function
log_string
(out_str)
tensorflow/sem_seg/batch_inference.py:34
↓ 3 callers
Method
_check_sanity
(self)
tensorflow/utils/plyfile.py:413
↓ 3 callers
Function
draw_point_cloud
Render point cloud to image with alpha channel. Input: points: Nx3 numpy array (+y is up direction) Output: g
tensorflow/utils/pc_util.py:97
↓ 3 callers
Method
read
Read PLY data from a readable file-like object or filename.
tensorflow/utils/plyfile.py:275
↓ 2 callers
Method
__init__
elements: sequence of PlyElement instances. text: whether the resulting PLY file will be text (True) or binary (False).
tensorflow/utils/plyfile.py:165
↓ 2 callers
Method
_index
(self)
tensorflow/utils/plyfile.py:428
↓ 2 callers
Function
_open_stream
(stream, read_or_write)
tensorflow/utils/plyfile.py:355
↓ 2 callers
Function
batch_norm_for_conv2d
Batch normalization on 2D convolutional maps. Args: inputs: Tensor, 4D BHWC input maps is_training: boolean tf.Varialbe, true i
tensorflow/utils/tf_util.py:577
↓ 2 callers
Function
convert_label_to_one_hot
(labels)
tensorflow/part_seg/train_multi_gpu.py:88
↓ 2 callers
Method
describe
Construct a PlyElement from an array's metadata. len_types and val_types can be given as mappings from list property names t
tensorflow/utils/plyfile.py:507
↓ 2 callers
Function
get_deconv_dim
(dim_size, stride_size, kernel_size, padding)
tensorflow/utils/tf_util.py:225
↓ 2 callers
Function
get_loss
pred: B,N,13; label: B,N
tensorflow/sem_seg/model.py:109
↓ 2 callers
Function
get_model
ConvNet baseline, input is BxNx9 gray image
tensorflow/sem_seg/model.py:20
↓ 2 callers
Function
input_transform_net
Input (XYZ) Transform Net, input is BxNx3 gray image Return: Transformation matrix of size 3xK
tensorflow/models/transform_nets.py:10
↓ 2 callers
Function
mat2euler
Discover Euler angle vector from 3x3 matrix Uses the conventions above. Parameters ---------- M : array-like, shape (3,3) cy_th
tensorflow/utils/eulerangles.py:198
↓ 2 callers
Function
output_color_point_cloud
(data, seg, out_file)
tensorflow/part_seg/test.py:60
↓ 2 callers
Function
placeholder_inputs
(batch_size, num_point)
tensorflow/sem_seg/model.py:13
↓ 2 callers
Function
room2blocks
Prepare block training data. Args: data: N x 6 numpy array, 012 are XYZ in meters, 345 are RGB in [0,1] assumes the data is shifted (min
tensorflow/sem_seg/indoor3d_util.py:133
↓ 1 callers
Method
__init__
(self, args, output_channels=40)
pytorch/model.py:89
↓ 1 callers
Method
_check_name
(self)
tensorflow/utils/plyfile.py:441
↓ 1 callers
Method
_check_name
(self)
tensorflow/utils/plyfile.py:741
↓ 1 callers
Method
_from_fields
Parse from generator. Raise StopIteration if the property could not be read.
tensorflow/utils/plyfile.py:778
↓ 1 callers
Method
_index
(self)
tensorflow/utils/plyfile.py:213
↓ 1 callers
Function
_init_
()
pytorch/main.py:27
↓ 1 callers
Method
_parse_header
Parse a PLY header from a readable file-like stream.
tensorflow/utils/plyfile.py:220
↓ 1 callers
Method
_parse_multi
Parse a list of PLY element definitions.
tensorflow/utils/plyfile.py:458
↓ 1 callers
Method
_parse_one
Consume one element definition. The unconsumed input is returned along with a PlyElement instance.
tensorflow/utils/plyfile.py:471
↓ 1 callers
Method
_read
Read the actual data from a PLY file.
tensorflow/utils/plyfile.py:567
↓ 1 callers
Method
_read_bin
Load a PLY element from a binary PLY file. The element may contain list properties.
tensorflow/utils/plyfile.py:656
↓ 1 callers
Method
_read_txt
Load a PLY element from an ASCII-format PLY file. The element may contain list properties.
tensorflow/utils/plyfile.py:611
↓ 1 callers
Function
_split_line
(line, n)
tensorflow/utils/plyfile.py:91
↓ 1 callers
Method
_to_fields
Return generator over one item.
tensorflow/utils/plyfile.py:786
↓ 1 callers
Method
_write
Write the data to a PLY file.
tensorflow/utils/plyfile.py:593
↓ 1 callers
Method
_write_bin
Save a PLY element to a binary PLY file. The element may contain list properties.
tensorflow/utils/plyfile.py:673
↓ 1 callers
Method
_write_txt
Save a PLY element to an ASCII-format PLY file. The element may contain list properties.
tensorflow/utils/plyfile.py:643
↓ 1 callers
Function
average_gradients
Calculate average gradient for each shared variable across all towers. Note that this function provides a synchronization point across all towers.
tensorflow/sem_seg/train.py:121
↓ 1 callers
Function
average_gradients
Calculate average gradient for each shared variable across all towers. Note that this function provides a synchronization point across all towers.
tensorflow/part_seg/train_multi_gpu.py:94
↓ 1 callers
Function
batch_norm_for_conv1d
Batch normalization on 1D convolutional maps. Args: inputs: Tensor, 3D BLC input maps is_training: boolean tf.Varialbe, true in
tensorflow/utils/tf_util.py:557
↓ 1 callers
Function
batch_norm_for_conv3d
Batch normalization on 3D convolutional maps. Args: inputs: Tensor, 5D BDHWC input maps is_training: boolean tf.Varialbe, true
tensorflow/utils/tf_util.py:596
↓ 1 callers
Function
batch_norm_for_fc
Batch normalization on FC data. Args: inputs: Tensor, 2D BxC input is_training: boolean tf.Varialbe, true indicates training ph
tensorflow/utils/tf_util.py:539
↓ 1 callers
Function
download
()
pytorch/data.py:19
↓ 1 callers
Function
euler2mat
Return matrix for rotations around z, y and x axes Uses the z, then y, then x convention above Parameters ---------- z : scalar
tensorflow/utils/eulerangles.py:98
↓ 1 callers
Function
euler2quat
Return quaternion corresponding to these Euler angles Uses the z, then y, then x convention above Parameters ---------- z : scalar
tensorflow/utils/eulerangles.py:271
↓ 1 callers
Function
eval_one_epoch
ops: dict mapping from string to tf ops
tensorflow/train.py:220
↓ 1 callers
Function
eval_one_epoch
(sess, ops, num_votes=1, topk=1)
tensorflow/evaluate.py:90
↓ 1 callers
Function
eval_one_epoch
(sess, ops, room_path, out_data_label_filename, out_gt_label_filename)
tensorflow/sem_seg/batch_inference.py:85
↓ 1 callers
Function
eval_one_epoch
(epoch_num)
tensorflow/part_seg/train_multi_gpu.py:300
↓ 1 callers
Function
evaluate
(num_votes)
tensorflow/evaluate.py:56
↓ 1 callers
Function
evaluate
()
tensorflow/sem_seg/batch_inference.py:39
↓ 1 callers
Function
get_bn_decay
(batch)
tensorflow/train.py:83
↓ 1 callers
Function
get_bn_decay
(batch)
tensorflow/sem_seg/train.py:111
↓ 1 callers
Function
get_learning_rate
(batch)
tensorflow/train.py:73
↓ 1 callers
Function
get_learning_rate
(batch)
tensorflow/sem_seg/train.py:101
↓ 1 callers
Function
get_model
Classification PointNet, input is BxNx3, output Bx40
tensorflow/models/dgcnn.py:20
↓ 1 callers
Function
knn
(x, k)
pytorch/model.py:21
↓ 1 callers
Function
load_data
(partition)
pytorch/data.py:32
↓ 1 callers
Function
load_h5
(h5_filename)
tensorflow/provider.py:142
↓ 1 callers
Function
load_pts_seg_files
(pts_file, seg_file, catid)
tensorflow/part_seg/test.py:101
↓ 1 callers
Function
output_color_point_cloud_red_blue
(data, seg, out_file)
tensorflow/part_seg/test.py:67
↓ 1 callers
Function
pc_augment_to_point_num
(pts, pn)
tensorflow/part_seg/test.py:110
↓ 1 callers
Function
pc_normalize
(pc)
tensorflow/part_seg/test.py:81
↓ 1 callers
Function
placeholder_inputs
()
tensorflow/part_seg/test.py:89
↓ 1 callers
Function
placeholder_inputs
(batch_size, num_point)
tensorflow/models/dgcnn.py:14
↓ 1 callers
Function
point_cloud_three_views
input points Nx3 numpy array (+y is up direction). return an numpy array gray image of size 500x1500.
tensorflow/utils/pc_util.py:156
↓ 1 callers
Function
point_cloud_three_views_demo
Demo for draw_point_cloud function
tensorflow/utils/pc_util.py:171
↓ 1 callers
Function
point_cloud_to_volume
input is Nx3 points. output is vsize*vsize*vsize assumes points are in range [-radius, radius]
tensorflow/utils/pc_util.py:41
↓ 1 callers
Function
predict
()
tensorflow/part_seg/test.py:125
↓ 1 callers
Function
pyplot_draw_point_cloud
points is a Nx3 numpy array
tensorflow/utils/pc_util.py:183
↓ 1 callers
Function
read_ply
read XYZ point cloud from filename PLY file
tensorflow/utils/pc_util.py:77
↓ 1 callers
Function
room2blocks_plus
room2block with input filename and RGB preprocessing.
tensorflow/sem_seg/indoor3d_util.py:205
↓ 1 callers
Function
room2blocks_plus_normalized
room2block, with input filename and RGB preprocessing. for each block centralize XYZ, add normalized XYZ as 678 channels
tensorflow/sem_seg/indoor3d_util.py:228
↓ 1 callers
Function
room2samples
Prepare whole room samples. Args: data: N x 6 numpy array, 012 are XYZ in meters, 345 are RGB in [0,1] assumes the data is shifted (min
tensorflow/sem_seg/indoor3d_util.py:267
↓ 1 callers
Function
room2samples_plus_normalized
room2sample, with input filename and RGB preprocessing. for each block centralize XYZ, add normalized XYZ as 678 channels
tensorflow/sem_seg/indoor3d_util.py:303
↓ 1 callers
Function
sample_data
data is in N x ... we want to keep num_samplexC of them. if N > num_sample, we will randomly keep num_sample of them. if N < num_sample,
tensorflow/sem_seg/indoor3d_util.py:111
↓ 1 callers
Function
sample_data_label
(data, label, num_sample)
tensorflow/sem_seg/indoor3d_util.py:128
↓ 1 callers
Function
test
(args, io)
pytorch/main.py:138
↓ 1 callers
Function
train
()
tensorflow/train.py:93
↓ 1 callers
Function
train
()
tensorflow/sem_seg/train.py:155
↓ 1 callers
Function
train
()
tensorflow/part_seg/train_multi_gpu.py:131
↓ 1 callers
Function
train
(args, io)
pytorch/main.py:39
↓ 1 callers
Function
train_one_epoch
ops: dict mapping from string to tf ops
tensorflow/train.py:171
↓ 1 callers
Function
train_one_epoch
ops: dict mapping from string to tf ops
tensorflow/sem_seg/train.py:243
↓ 1 callers
Function
train_one_epoch
(train_file_idx, epoch_num)
tensorflow/part_seg/train_multi_gpu.py:234
next →
1–100 of 215, ranked by callers