MCPcopy Create free account

hub / github.com/LZDSJTU/pointnet_pytorch / functions

Functions62 in github.com/LZDSJTU/pointnet_pytorch

↓ 8 callersMethodupdate
(self, val, n=1)
utils/train_utils.py:16
↓ 3 callersFunctionaccuracy
Computes the precision@k for the specified values of k
utils/train_utils.py:22
↓ 2 callersFunctionroom2blocks
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 s
utils/indoor3d_util.py:132
↓ 2 callersFunctionroom2blocks
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 s
data_preparation/indoor3d_util.py:134
↓ 1 callersFunctionadjust_learning_rate
(optimizer, global_counter, batch_size, base_lr)
utils/train_utils.py:38
↓ 1 callersFunctionevaluate
(room_path, out_data_label_filename, out_gt_label_filename)
inference_ptn.py:50
↓ 1 callersFunctiongetDataFiles
(list_filename)
utils/train_utils.py:68
↓ 1 callersFunctionget_model
()
inference_ptn.py:45
↓ 1 callersFunctionget_model
()
train_ptn.py:189
↓ 1 callersFunctioninsert_batch
(data, label, last_batch=False)
data_preparation/gen_indoor3d_h5.py:42
↓ 1 callersFunctionloadDataFile
(filename)
utils/train_utils.py:62
↓ 1 callersFunctionmain
()
train_ptn.py:19
↓ 1 callersMethodreset
(self)
utils/train_utils.py:10
↓ 1 callersFunctionroom2blocks_plus
room2block with input filename and RGB preprocessing.
utils/indoor3d_util.py:204
↓ 1 callersFunctionroom2blocks_plus
room2block with input filename and RGB preprocessing.
data_preparation/indoor3d_util.py:206
↓ 1 callersFunctionroom2blocks_plus_normalized
room2block, with input filename and RGB preprocessing. for each block centralize XYZ, add normalized XYZ as 678 channels
utils/indoor3d_util.py:227
↓ 1 callersFunctionroom2blocks_plus_normalized
room2block, with input filename and RGB preprocessing. for each block centralize XYZ, add normalized XYZ as 678 channels
data_preparation/indoor3d_util.py:229
↓ 1 callersFunctionroom2blocks_wrapper_normalized
(data_label_filename, num_point, block_size=1.0, stride=1.0, random_sample=
utils/indoor3d_util.py:254
↓ 1 callersFunctionroom2samples
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 s
utils/indoor3d_util.py:266
↓ 1 callersFunctionroom2samples
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 s
data_preparation/indoor3d_util.py:268
↓ 1 callersFunctionroom2samples_plus_normalized
room2sample, with input filename and RGB preprocessing. for each block centralize XYZ, add normalized XYZ as 678 channels
utils/indoor3d_util.py:302
↓ 1 callersFunctionroom2samples_plus_normalized
room2sample, with input filename and RGB preprocessing. for each block centralize XYZ, add normalized XYZ as 678 channels
data_preparation/indoor3d_util.py:304
↓ 1 callersFunctionsample_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 <
utils/indoor3d_util.py:110
↓ 1 callersFunctionsample_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 <
data_preparation/indoor3d_util.py:110
↓ 1 callersFunctionsample_data_label
(data, label, num_sample)
utils/indoor3d_util.py:127
↓ 1 callersFunctionsample_data_label
(data, label, num_sample)
data_preparation/indoor3d_util.py:129
↓ 1 callersFunctionshuffle_data
Shuffle data and labels. Input: data: B,N,... numpy array label: B,... numpy array Return: shuffled dat
utils/train_utils.py:50
Method__init__
(self)
utils/train_utils.py:7
Method__init__
(self)
model/pointnet.py:8
Functionadjust_bn_decay
Sets the learning rate to the initial LR decayed by 10 every 30 epochs
utils/train_utils.py:44
Functionbatch_mkdir
(output_folder, subdir_list)
data_preparation/data_prep_util.py:48
Functionbbox_label_to_obj
Visualization of bounding boxes. Args: input_filename: each line is x1 y1 z1 x2 y2 z2 label out_filename_prefix: OBJ filenam
utils/indoor3d_util.py:385
Functionbbox_label_to_obj
Visualization of bounding boxes. Args: input_filename: each line is x1 y1 z1 x2 y2 z2 label out_filename_prefix: OBJ filenam
data_preparation/indoor3d_util.py:387
Functionbbox_label_to_obj_room
Visualization of bounding boxes. Args: input_filename: each line is x1 y1 z1 x2 y2 z2 label out_filename_prefix: OBJ filenam
utils/indoor3d_util.py:448
Functionbbox_label_to_obj_room
Visualization of bounding boxes. Args: input_filename: each line is x1 y1 z1 x2 y2 z2 label out_filename_prefix: OBJ filenam
data_preparation/indoor3d_util.py:450
Functioncollect_bounding_box
Compute bounding boxes from each instance in original dataset files on one room. **We assume the bbox is aligned with XYZ coordinate.**
utils/indoor3d_util.py:343
Functioncollect_bounding_box
Compute bounding boxes from each instance in original dataset files on one room. **We assume the bbox is aligned with XYZ coordinate.**
data_preparation/indoor3d_util.py:345
Functioncollect_point_bounding_box
Compute bounding boxes from each instance in original dataset files on one room. **We assume the bbox is aligned with XYZ coordinate.**
utils/indoor3d_util.py:527
Functioncollect_point_bounding_box
Compute bounding boxes from each instance in original dataset files on one room. **We assume the bbox is aligned with XYZ coordinate.**
data_preparation/indoor3d_util.py:529
Functioncollect_point_label
Convert original dataset files to data_label file (each line is XYZRGBL). We aggregated all the points from each instance in the room. A
utils/indoor3d_util.py:37
Functioncollect_point_label
Convert original dataset files to data_label file (each line is XYZRGBL). We aggregated all the points from each instance in the room. A
data_preparation/indoor3d_util.py:37
Functionexport_ply
(pc, filename)
data_preparation/data_prep_util.py:15
Methodforward
(self, input)
model/pointnet.py:54
Functionget_category_names
()
data_preparation/data_prep_util.py:35
Functionget_obj_filenames
()
data_preparation/data_prep_util.py:41
Functionget_sampling_command
(obj_filename, ply_filename)
data_preparation/data_prep_util.py:23
Functionload_h5
(h5_filename)
data_preparation/data_prep_util.py:125
Functionload_h5_data_label_normal
(h5_filename)
data_preparation/data_prep_util.py:109
Functionload_h5_data_label_seg
(h5_filename)
data_preparation/data_prep_util.py:117
Functionload_ply_data
(filename, point_num)
data_preparation/data_prep_util.py:136
Functionload_ply_normal
(filename, point_num)
data_preparation/data_prep_util.py:143
Functionpad_arr_rows
(arr, row, pad='edge')
data_preparation/data_prep_util.py:151
Functionpoint_label_to_obj
For visualization of a room from data_label file, input_filename: each line is X Y Z R G B L out_filename: OBJ filename, visualize inpu
utils/indoor3d_util.py:79
Functionpoint_label_to_obj
For visualization of a room from data_label file, input_filename: each line is X Y Z R G B L out_filename: OBJ filename, visualize inpu
data_preparation/indoor3d_util.py:79
Functionroom2blocks_wrapper
(data_label_filename, num_point, block_size=1.0, stride=1.0, random_sample=False, samp
utils/indoor3d_util.py:215
Functionroom2blocks_wrapper
(data_label_filename, num_point, block_size=1.0, stride=1.0, random_sample=False, samp
data_preparation/indoor3d_util.py:217
Functionroom2blocks_wrapper_normalized
(data_label_filename, num_point, block_size=1.0, stride=1.0, random_sample=
data_preparation/indoor3d_util.py:256
Functionroom2samples_wrapper_normalized
(data_label_filename, num_point)
utils/indoor3d_util.py:328
Functionroom2samples_wrapper_normalized
(data_label_filename, num_point)
data_preparation/indoor3d_util.py:330
Functionsave_h5
(h5_filename, data, label, data_dtype='uint8', label_dtype='uint8')
data_preparation/data_prep_util.py:79
Functionsave_h5_addcoordinate
(h5_filename, data, label, coordinate, data_dtype='uint8', label_dtype='uint8', coordinate_dtype='uint8')
data_preparation/data_prep_util.py:92
Functionsave_h5_data_label_normal
(h5_filename, data, label, normal, data_dtype='float32', label_dtype='uint8', noral_dtype='float32')
data_preparation/data_prep_util.py:60