SYMBOL INDEX (478 symbols across 74 files) FILE: OcCo_TF/cls_models/dgcnn_cls.py class Model (line 10) | class Model: method __init__ (line 11) | def __init__(self, inputs, npts, labels, is_training, **kwargs): method get_graph_feature (line 20) | def get_graph_feature(x, k): method create_encoder (line 27) | def create_encoder(self, point_cloud): method create_decoder (line 75) | def create_decoder(self, features): method create_loss (line 98) | def create_loss(pred, label, smoothing=True): FILE: OcCo_TF/cls_models/pcn_cls.py class Model (line 10) | class Model: method __init__ (line 11) | def __init__(self, inputs, npts, labels, is_training, **kwargs): method create_encoder (line 17) | def create_encoder(self, inputs, npts): method create_decoder (line 29) | def create_decoder(self, features): method create_loss (line 43) | def create_loss(self, pred, label): FILE: OcCo_TF/cls_models/pointnet_cls.py class Model (line 13) | class Model: method __init__ (line 14) | def __init__(self, inputs, npts, labels, is_training, **kwargs): method create_encoder (line 21) | def create_encoder(self, inputs, npts): method create_decoder (line 65) | def create_decoder(self, features): method create_loss (line 79) | def create_loss(self, pred, label): FILE: OcCo_TF/completion_models/dgcnn_cd.py class Model (line 18) | class Model: method __init__ (line 19) | def __init__(self, inputs, npts, gt, alpha, **kwargs): method create_encoder (line 34) | def create_encoder(self, point_cloud, npts): method create_decoder (line 100) | def create_decoder(self, features): method create_loss (line 123) | def create_loss(self, gt, alpha): FILE: OcCo_TF/completion_models/dgcnn_emd.py class Model (line 18) | class Model: method __init__ (line 19) | def __init__(self, inputs, npts, gt, alpha, **kwargs): method create_encoder (line 34) | def create_encoder(self, point_cloud, npts): method create_decoder (line 100) | def create_decoder(self, features): method create_loss (line 123) | def create_loss(self, gt, alpha): FILE: OcCo_TF/completion_models/pcn_cd.py class Model (line 9) | class Model: method __init__ (line 10) | def __init__(self, inputs, npts, gt, alpha, **kwargs): method create_encoder (line 23) | def create_encoder(self, inputs, npts): method create_decoder (line 33) | def create_decoder(self, features): method create_loss (line 56) | def create_loss(self, coarse, fine, gt, alpha): FILE: OcCo_TF/completion_models/pcn_emd.py class Model (line 10) | class Model: method __init__ (line 11) | def __init__(self, inputs, npts, gt, alpha, **kwargs): method create_encoder (line 23) | def create_encoder(self, inputs, npts): method create_decoder (line 33) | def create_decoder(self, features): method create_loss (line 58) | def create_loss(self, coarse, fine, gt, alpha): FILE: OcCo_TF/completion_models/pointnet_cd.py class Model (line 13) | class Model: method __init__ (line 14) | def __init__(self, inputs, npts, gt, alpha, **kwargs): method create_encoder (line 28) | def create_encoder(self, inputs, npts): method create_decoder (line 82) | def create_decoder(self, features): method create_loss (line 106) | def create_loss(self, gt, alpha): FILE: OcCo_TF/completion_models/pointnet_emd.py class Model (line 16) | class Model: method __init__ (line 17) | def __init__(self, inputs, npts, gt, alpha, **kwargs): method create_encoder (line 31) | def create_encoder(self, inputs, npts): method create_decoder (line 83) | def create_decoder(self, features): method create_loss (line 106) | def create_loss(self, gt, alpha): FILE: OcCo_TF/pc_distance/tf_approxmatch.cpp function approxmatch_cpu (line 23) | void approxmatch_cpu(int b,int n,int m,const float * xyz1,const float * ... function matchcost_cpu (line 85) | void matchcost_cpu(int b,int n,int m,const float * xyz1,const float * xy... function matchcostgrad_cpu (line 106) | void matchcostgrad_cpu(int b,int n,int m,const float * xyz1,const float ... class ApproxMatchGpuOp (line 145) | class ApproxMatchGpuOp: public OpKernel{ method ApproxMatchGpuOp (line 147) | explicit ApproxMatchGpuOp(OpKernelConstruction* context):OpKernel(cont... method Compute (line 148) | void Compute(OpKernelContext * context)override{ class ApproxMatchOp (line 175) | class ApproxMatchOp: public OpKernel{ method ApproxMatchOp (line 177) | explicit ApproxMatchOp(OpKernelConstruction* context):OpKernel(context){} method Compute (line 178) | void Compute(OpKernelContext * context)override{ class MatchCostGpuOp (line 201) | class MatchCostGpuOp: public OpKernel{ method MatchCostGpuOp (line 203) | explicit MatchCostGpuOp(OpKernelConstruction* context):OpKernel(contex... method Compute (line 204) | void Compute(OpKernelContext * context)override{ class MatchCostOp (line 231) | class MatchCostOp: public OpKernel{ method MatchCostOp (line 233) | explicit MatchCostOp(OpKernelConstruction* context):OpKernel(context){} method Compute (line 234) | void Compute(OpKernelContext * context)override{ class MatchCostGradGpuOp (line 262) | class MatchCostGradGpuOp: public OpKernel{ method MatchCostGradGpuOp (line 264) | explicit MatchCostGradGpuOp(OpKernelConstruction* context):OpKernel(co... method Compute (line 265) | void Compute(OpKernelContext * context)override{ class MatchCostGradOp (line 296) | class MatchCostGradOp: public OpKernel{ method MatchCostGradOp (line 298) | explicit MatchCostGradOp(OpKernelConstruction* context):OpKernel(conte... method Compute (line 299) | void Compute(OpKernelContext * context)override{ FILE: OcCo_TF/pc_distance/tf_approxmatch.py function approx_match (line 11) | def approx_match(xyz1, xyz2): function _approx_match_shape (line 25) | def _approx_match_shape(op): function match_cost (line 31) | def match_cost(xyz1, xyz2, match): function _match_cost_shape (line 43) | def _match_cost_shape(op): function _match_cost_grad (line 51) | def _match_cost_grad(op,grad_cost): FILE: OcCo_TF/pc_distance/tf_nndistance.cpp function nnsearch (line 21) | static void nnsearch(int b,int n,int m,const float * xyz1,const float * ... class NnDistanceOp (line 45) | class NnDistanceOp : public OpKernel{ method NnDistanceOp (line 47) | explicit NnDistanceOp(OpKernelConstruction* context):OpKernel(context){} method Compute (line 48) | void Compute(OpKernelContext * context)override{ class NnDistanceGradOp (line 84) | class NnDistanceGradOp : public OpKernel{ method NnDistanceGradOp (line 86) | explicit NnDistanceGradOp(OpKernelConstruction* context):OpKernel(cont... method Compute (line 87) | void Compute(OpKernelContext * context)override{ class NnDistanceGpuOp (line 169) | class NnDistanceGpuOp : public OpKernel{ method NnDistanceGpuOp (line 171) | explicit NnDistanceGpuOp(OpKernelConstruction* context):OpKernel(conte... method Compute (line 172) | void Compute(OpKernelContext * context)override{ class NnDistanceGradGpuOp (line 209) | class NnDistanceGradGpuOp : public OpKernel{ method NnDistanceGradGpuOp (line 211) | explicit NnDistanceGradGpuOp(OpKernelConstruction* context):OpKernel(c... method Compute (line 212) | void Compute(OpKernelContext * context)override{ FILE: OcCo_TF/pc_distance/tf_nndistance.py function nn_distance (line 10) | def nn_distance(xyz1, xyz2): function _nn_distance_grad (line 24) | def _nn_distance_grad(op, grad_dist1, grad_idx1, grad_dist2, grad_idx2): FILE: OcCo_TF/train_cls.py function log_string (line 62) | def log_string(out_str): function train (line 68) | def train(args): function train_one_epoch (line 186) | def train_one_epoch(sess, ops, train_writer): function eval_one_epoch (line 234) | def eval_one_epoch(sess, ops, val_writer): FILE: OcCo_TF/train_cls_dgcnn_torchloader.py function parse_args (line 15) | def parse_args(): function main (line 56) | def main(args): function train_one_epoch (line 174) | def train_one_epoch(sess, ops, MyLogger, train_writer): function eval_one_epoch (line 205) | def eval_one_epoch(sess, ops, MyLogger, val_writer): FILE: OcCo_TF/train_cls_torchloader.py function log_string (line 70) | def log_string(out_str): function train (line 76) | def train(args): function train_one_epoch (line 195) | def train_one_epoch(sess, ops, train_writer): function eval_one_epoch (line 268) | def eval_one_epoch(sess, ops, val_writer): FILE: OcCo_TF/train_completion.py function get_bn_decay (line 48) | def get_bn_decay(batch): function vary2fix (line 59) | def vary2fix(inputs, npts): function train (line 77) | def train(args): FILE: OcCo_TF/utils/Dataset_Assign.py function Dataset_Assign (line 5) | def Dataset_Assign(dataset, fname, partial=True, bn=False, few_shot=False): FILE: OcCo_TF/utils/EarlyStoppingCriterion.py class EarlyStoppingCriterion (line 3) | class EarlyStoppingCriterion(object): method __init__ (line 12) | def __init__(self, patience=10, mode='max', min_delta=0.0): method step (line 26) | def step(self, cur_dev_score, epoch): FILE: OcCo_TF/utils/ModelNetDataLoader.py function pc_normalize (line 7) | def pc_normalize(pc): function farthest_point_sample (line 15) | def farthest_point_sample(point, npoint): class ModelNetDataLoader (line 39) | class ModelNetDataLoader(Dataset): method __init__ (line 40) | def __init__(self, root, npoint=1024, split='train', uniform=False, no... method __len__ (line 63) | def __len__(self): method _get_item (line 66) | def _get_item(self, index): method __getitem__ (line 89) | def __getitem__(self, index): class General_CLSDataLoader_HDF5 (line 93) | class General_CLSDataLoader_HDF5(Dataset): method __init__ (line 94) | def __init__(self, file_list, num_point=1024): method loadh5DataFile (line 116) | def loadh5DataFile(PathtoFile): method __len__ (line 120) | def __len__(self): method __getitem__ (line 123) | def __getitem__(self, index): class ModelNetJigsawDataLoader (line 131) | class ModelNetJigsawDataLoader(Dataset): method __init__ (line 132) | def __init__(self, root=r'./data/modelnet40_ply_hdf5_2048/jigsaw', method loadh5DataFile (line 164) | def loadh5DataFile(PathtoFile): method __getitem__ (line 168) | def __getitem__(self, index): method __len__ (line 177) | def __len__(self): FILE: OcCo_TF/utils/Train_Logger.py class TrainLogger (line 7) | class TrainLogger: method __init__ (line 9) | def __init__(self, args, name='Model', subfold='cls', cls2name=None): method logger_setup (line 24) | def logger_setup(self): method make_logdir (line 42) | def make_logdir(self): method epoch_init (line 62) | def epoch_init(self, training=True): method step_update (line 67) | def step_update(self, pred, gt, loss, training=True): method cls_epoch_update (line 76) | def cls_epoch_update(self, training=True): method seg_epoch_update (line 95) | def seg_epoch_update(self, training=True): method epoch_summary (line 109) | def epoch_summary(self, writer=None, training=True): method train_summary (line 129) | def train_summary(self): method update_from_checkpoints (line 136) | def update_from_checkpoints(self, checkpoint): method update_from_checkpoints_tf (line 145) | def update_from_checkpoints_tf(self, checkpoint): FILE: OcCo_TF/utils/check_scale.py function log_string (line 9) | def log_string(msg): FILE: OcCo_TF/utils/data_util.py function resample_pcd (line 8) | def resample_pcd(pcd, n): class PreprocessData (line 16) | class PreprocessData(dataflow.ProxyDataFlow): method __init__ (line 17) | def __init__(self, ds, input_size, output_size): method get_data (line 23) | def get_data(self): class BatchData (line 30) | class BatchData(dataflow.ProxyDataFlow): method __init__ (line 31) | def __init__(self, ds, batch_size, input_size, gt_size, remainder=Fals... method __len__ (line 39) | def __len__(self): method __iter__ (line 48) | def __iter__(self): method _aggregate_batch (line 59) | def _aggregate_batch(self, data_holder, use_list=False): function lmdb_dataflow (line 74) | def lmdb_dataflow(lmdb_path, batch_size, input_size, output_size, is_tra... function get_queued_data (line 91) | def get_queued_data(generator, dtypes, shapes, queue_capacity=10): FILE: OcCo_TF/utils/io_util.py function read_pcd (line 9) | def read_pcd(filename): function save_pcd (line 14) | def save_pcd(filename, points): function shuffle_data (line 20) | def shuffle_data(data, labels): function loadh5DataFile (line 27) | def loadh5DataFile(PathtoFile): function getDataFiles (line 32) | def getDataFiles(list_filename): function save_h5 (line 36) | def save_h5(h5_filename, data, label, data_dtype='uint8', label_dtype='u... FILE: OcCo_TF/utils/pc_util.py function jitter_point_cloud (line 6) | def jitter_point_cloud(batch_data, sigma=0.01, clip=0.05): function rotate_point_cloud (line 20) | def rotate_point_cloud(batch_data): function rotate_point_cloud_by_angle (line 41) | def rotate_point_cloud_by_angle(batch_data, rotation_angle): function random_point_dropout (line 61) | def random_point_dropout(batch_pc, max_dropout_ratio=0.875): function random_scale_point_cloud (line 72) | def random_scale_point_cloud(batch_data, scale_low=0.8, scale_high=1.25): function random_shift_point_cloud (line 86) | def random_shift_point_cloud(batch_data, shift_range=0.1): FILE: OcCo_TF/utils/tf_util.py function mlp (line 12) | def mlp(features, layer_dims, bn=None, bn_params=None): function mlp_conv (line 27) | def mlp_conv(inputs, layer_dims, bn=None, bn_params=None): function point_maxpool (line 46) | def point_maxpool(inputs, npts, keepdims=False): function point_unpool (line 52) | def point_unpool(inputs, npts): function chamfer (line 58) | def chamfer(pcd1, pcd2): function earth_mover (line 66) | def earth_mover(pcd1, pcd2): function add_train_summary (line 75) | def add_train_summary(name, value): function add_valid_summary (line 79) | def add_valid_summary(name, value): function _variable_on_cpu (line 88) | def _variable_on_cpu(name, shape, initializer, use_fp16=False): function _variable_with_weight_decay (line 104) | def _variable_with_weight_decay(name, shape, stddev, wd, use_xavier=True): function batch_norm_template (line 132) | def batch_norm_template(inputs, is_training, scope, moments_dims, bn_dec... function batch_norm_for_fc (line 176) | def batch_norm_for_fc(inputs, is_training, bn_decay, scope): function fully_connected (line 190) | def fully_connected(inputs, function max_pool2d (line 231) | def max_pool2d(inputs, function dropout (line 264) | def dropout(inputs, function conv2d (line 288) | def conv2d(inputs, function batch_norm_for_conv2d (line 364) | def batch_norm_for_conv2d(inputs, is_training, bn_decay, scope): function batch_norm_template (line 369) | def batch_norm_template(inputs, is_training, scope, moments_dims, bn_dec... function pairwise_distance (line 417) | def pairwise_distance(point_cloud): function knn (line 439) | def knn(adj_matrix, k=20): function get_edge_feature (line 453) | def get_edge_feature(point_cloud, nn_idx, k=20): function get_learning_rate (line 489) | def get_learning_rate(batch, base_lr, batch_size, decay_step, decay_rate... function get_lr_dgcnn (line 500) | def get_lr_dgcnn(batch, base_lr, batch_size, decay_step, alpha): function get_bn_decay (line 509) | def get_bn_decay(batch, bn_init_decay, batch_size, bn_decay_step, bn_dec... FILE: OcCo_TF/utils/transfer_pretrained_w.py function load_para_from_saved_model (line 7) | def load_para_from_saved_model(model_path, verbose=False): function intersec_saved_var (line 24) | def intersec_saved_var(model_path1, model_path2, verbose=False): function load_pretrained_var (line 40) | def load_pretrained_var(source_model_path, target_model_path, verbose=Fa... FILE: OcCo_TF/utils/transform_nets.py function input_transform_net_dgcnn (line 7) | def input_transform_net_dgcnn(edge_feature, is_training, bn_decay=None, ... function input_transform_net (line 56) | def input_transform_net(point_cloud, is_training, bn_decay=None, K=3): function feature_transform_net (line 113) | def feature_transform_net(inputs, is_training, bn_decay=None, K=64): FILE: OcCo_TF/utils/visu_util.py function plot_pcd_three_views (line 11) | def plot_pcd_three_views(filename, pcds, titles, suptitle='', sizes=None... FILE: OcCo_Torch/chamfer_distance/chamfer_distance.cpp function chamfer_distance_forward_cuda (line 27) | void chamfer_distance_forward_cuda( function chamfer_distance_backward_cuda (line 41) | void chamfer_distance_backward_cuda( function nnsearch (line 59) | void nnsearch( function chamfer_distance_forward (line 90) | void chamfer_distance_forward( function chamfer_distance_backward (line 114) | void chamfer_distance_backward( function PYBIND11_MODULE (line 180) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) { FILE: OcCo_Torch/chamfer_distance/chamfer_distance.py class ChamferDistanceFunction (line 10) | class ChamferDistanceFunction(torch.autograd.Function): method forward (line 12) | def forward(ctx, xyz1, xyz2): method backward (line 37) | def backward(ctx, graddist1, graddist2): class ChamferDistance (line 56) | class ChamferDistance(nn.Module): method forward (line 57) | def forward(self, xyz1, xyz2): class get_model (line 61) | class get_model(nn.Module): method __init__ (line 62) | def __init__(self, channel=3): method forward (line 67) | def forward(self, x): FILE: OcCo_Torch/models/dgcnn_cls.py class get_model (line 7) | class get_model(nn.Module): method __init__ (line 9) | def __init__(self, args, num_channel=3, num_class=40, **kwargs): method forward (line 42) | def forward(self, x): class get_loss (line 75) | class get_loss(torch.nn.Module): method __init__ (line 76) | def __init__(self): method cal_loss (line 80) | def cal_loss(pred, gold, smoothing=True): method forward (line 96) | def forward(self, pred, target): FILE: OcCo_Torch/models/dgcnn_jigsaw.py class get_model (line 8) | class get_model(nn.Module): method __init__ (line 9) | def __init__(self, args, num_class, **kwargs): method forward (line 50) | def forward(self, x): class get_loss (line 89) | class get_loss(torch.nn.Module): method __init__ (line 90) | def __init__(self): method cal_loss (line 94) | def cal_loss(pred, gold, smoothing=False): method forward (line 111) | def forward(self, pred, target): FILE: OcCo_Torch/models/dgcnn_occo.py class get_model (line 11) | class get_model(nn.Module): method __init__ (line 12) | def __init__(self, **kwargs): method build_grid (line 61) | def build_grid(self, batch_size): method tile (line 69) | def tile(self, tensor, multiples): method expand_dims (line 89) | def expand_dims(tensor, dim): method forward (line 94) | def forward(self, x): class get_loss (line 140) | class get_loss(nn.Module): method __init__ (line 141) | def __init__(self): method dist_cd (line 145) | def dist_cd(pc1, pc2): method forward (line 150) | def forward(self, coarse, fine, gt, alpha): FILE: OcCo_Torch/models/dgcnn_partseg.py class get_model (line 9) | class get_model(nn.Module): method __init__ (line 10) | def __init__(self, args, part_num=50, num_channel=3, **kwargs): method forward (line 61) | def forward(self, x, l): class get_loss (line 111) | class get_loss(nn.Module): method __init__ (line 112) | def __init__(self): method cal_loss (line 116) | def cal_loss(pred, gold, smoothing=False): method forward (line 133) | def forward(self, pred, target): FILE: OcCo_Torch/models/dgcnn_semseg.py class get_model (line 9) | class get_model(nn.Module): method __init__ (line 10) | def __init__(self, args, num_class, num_channel=9, **kwargs): method forward (line 50) | def forward(self, x): class get_loss (line 83) | class get_loss(nn.Module): method __init__ (line 84) | def __init__(self): method cal_loss (line 88) | def cal_loss(pred, gold, smoothing=False): method forward (line 105) | def forward(self, pred, target): FILE: OcCo_Torch/models/dgcnn_util.py function knn (line 6) | def knn(x, k): function get_graph_feature (line 14) | def get_graph_feature(x, k=20, idx=None, extra_dim=False): class T_Net (line 38) | class T_Net(nn.Module): method __init__ (line 41) | def __init__(self, channel=3, k=3): method forward (line 67) | def forward(self, x): class encoder (line 86) | class encoder(nn.Module): method __init__ (line 87) | def __init__(self, channel=3, **kwargs): method forward (line 111) | def forward(self, x): FILE: OcCo_Torch/models/pcn_cls.py class get_model (line 6) | class get_model(nn.Module): method __init__ (line 7) | def __init__(self, num_class=40, num_channel=3, **kwargs): method forward (line 19) | def forward(self, x): class get_loss (line 32) | class get_loss(nn.Module): method __init__ (line 33) | def __init__(self): method forward (line 36) | def forward(self, pred, target): FILE: OcCo_Torch/models/pcn_jigsaw.py class get_model (line 7) | class get_model(nn.Module): method __init__ (line 8) | def __init__(self, num_class, num_channel=3, **kwargs): method forward (line 20) | def forward(self, x): class get_loss (line 33) | class get_loss(nn.Module): method __init__ (line 34) | def __init__(self): method forward (line 37) | def forward(self, pred, target, trans_feat, weight): FILE: OcCo_Torch/models/pcn_occo.py class get_model (line 12) | class get_model(nn.Module): method __init__ (line 13) | def __init__(self, **kwargs): method build_grid (line 47) | def build_grid(self, batch_size): method tile (line 55) | def tile(self, tensor, multiples): method expand_dims (line 75) | def expand_dims(tensor, dim): method forward (line 81) | def forward(self, x): class get_loss (line 106) | class get_loss(nn.Module): method __init__ (line 107) | def __init__(self): method dist_cd (line 111) | def dist_cd(pc1, pc2): method forward (line 116) | def forward(self, coarse, fine, gt, alpha): FILE: OcCo_Torch/models/pcn_partseg.py class get_model (line 7) | class get_model(nn.Module): method __init__ (line 8) | def __init__(self, part_num=50, num_channel=3, **kwargs): method forward (line 21) | def forward(self, point_cloud, label): class get_loss (line 33) | class get_loss(nn.Module): method __init__ (line 34) | def __init__(self): method forward (line 37) | def forward(self, pred, target): FILE: OcCo_Torch/models/pcn_semseg.py class get_model (line 7) | class get_model(nn.Module): method __init__ (line 8) | def __init__(self, num_class, num_channel=9, **kwargs): method forward (line 20) | def forward(self, x): class get_loss (line 33) | class get_loss(nn.Module): method __init__ (line 34) | def __init__(self): method forward (line 37) | def forward(self, pred, target): FILE: OcCo_Torch/models/pcn_util.py class PCNEncoder (line 5) | class PCNEncoder(nn.Module): method __init__ (line 6) | def __init__(self, global_feat=False, channel=3): method forward (line 16) | def forward(self, x): class PCNPartSegEncoder (line 38) | class PCNPartSegEncoder(nn.Module): method __init__ (line 39) | def __init__(self, channel=3): method forward (line 47) | def forward(self, x, label): class encoder (line 70) | class encoder(nn.Module): method __init__ (line 71) | def __init__(self, num_channel=3, **kwargs): method forward (line 75) | def forward(self, x): FILE: OcCo_Torch/models/pointnet_cls.py class get_model (line 8) | class get_model(nn.Module): method __init__ (line 9) | def __init__(self, num_class=40, num_channel=3, **kwargs): method forward (line 20) | def forward(self, x): class get_loss (line 29) | class get_loss(nn.Module): method __init__ (line 30) | def __init__(self, mat_diff_loss_scale=0.001): method forward (line 34) | def forward(self, pred, target, trans_feat): FILE: OcCo_Torch/models/pointnet_jigsaw.py class get_model (line 7) | class get_model(nn.Module): method __init__ (line 8) | def __init__(self, num_class, num_channel=3, **kwargs): method forward (line 22) | def forward(self, x): class get_loss (line 35) | class get_loss(nn.Module): method __init__ (line 36) | def __init__(self, mat_diff_loss_scale=0.001): method forward (line 40) | def forward(self, pred, target, trans_feat): FILE: OcCo_Torch/models/pointnet_occo.py class get_model (line 14) | class get_model(nn.Module): method __init__ (line 15) | def __init__(self, **kwargs): method build_grid (line 43) | def build_grid(self, batch_size): method tile (line 51) | def tile(self, tensor, multiples): method expand_dims (line 71) | def expand_dims(tensor, dim): method forward (line 76) | def forward(self, x): class get_loss (line 99) | class get_loss(nn.Module): method __init__ (line 100) | def __init__(self): method dist_cd (line 104) | def dist_cd(pc1, pc2): method forward (line 109) | def forward(self, coarse, fine, gt, alpha): FILE: OcCo_Torch/models/pointnet_partseg.py class get_model (line 8) | class get_model(nn.Module): method __init__ (line 9) | def __init__(self, part_num=50, num_channel=3, **kwargs): method forward (line 23) | def forward(self, point_cloud, label): class get_loss (line 37) | class get_loss(nn.Module): method __init__ (line 38) | def __init__(self, mat_diff_loss_scale=0.001): method forward (line 42) | def forward(self, pred, target, trans_feat): FILE: OcCo_Torch/models/pointnet_semseg.py class get_model (line 7) | class get_model(nn.Module): method __init__ (line 8) | def __init__(self, num_class=13, num_channel=9, **kwargs): method forward (line 23) | def forward(self, x): class get_loss (line 36) | class get_loss(nn.Module): method __init__ (line 37) | def __init__(self, mat_diff_loss_scale=0.001): method forward (line 41) | def forward(self, pred, target, trans_feat): FILE: OcCo_Torch/models/pointnet_util.py function feature_transform_regularizer (line 8) | def feature_transform_regularizer(trans): class STN3d (line 18) | class STN3d(nn.Module): method __init__ (line 19) | def __init__(self, channel): method forward (line 35) | def forward(self, x): class STNkd (line 54) | class STNkd(nn.Module): method __init__ (line 55) | def __init__(self, k=64): method forward (line 73) | def forward(self, x): class PointNetEncoder (line 93) | class PointNetEncoder(nn.Module): method __init__ (line 94) | def __init__(self, global_feat=True, feature_transform=False, method forward (line 111) | def forward(self, x): class PointNetPartSegEncoder (line 153) | class PointNetPartSegEncoder(nn.Module): method __init__ (line 154) | def __init__(self, feature_transform=True, channel=3): method forward (line 173) | def forward(self, point_cloud, label): class encoder (line 207) | class encoder(nn.Module): method __init__ (line 208) | def __init__(self, num_channel=3, **kwargs): method forward (line 212) | def forward(self, x): class detailed_encoder (line 217) | class detailed_encoder(nn.Module): method __init__ (line 218) | def __init__(self, num_channel=3, **kwargs): method forward (line 224) | def forward(self, x): FILE: OcCo_Torch/train_cls.py function parse_args (line 20) | def parse_args(): function main (line 54) | def main(args): FILE: OcCo_Torch/train_completion.py function parse_args (line 18) | def parse_args(): function main (line 50) | def main(args): FILE: OcCo_Torch/train_jigsaw.py function parse_args (line 17) | def parse_args(): function main (line 43) | def main(args): FILE: OcCo_Torch/train_partseg.py function parse_args (line 39) | def parse_args(): function main (line 68) | def main(args): FILE: OcCo_Torch/train_semseg.py function parse_args (line 28) | def parse_args(): function main (line 55) | def main(args): FILE: OcCo_Torch/train_svm.py function parse_args (line 18) | def parse_args(): FILE: OcCo_Torch/utils/3DPC_Data_Gen.py function pc_ssl_3djigsaw_gen (line 12) | def pc_ssl_3djigsaw_gen(pc_xyz, k=2, edge_len=1): function loadh5DataFile (line 60) | def loadh5DataFile(PathtoFile): function reduce2fix (line 65) | def reduce2fix(pc, n_points=1024): FILE: OcCo_Torch/utils/Dataset_Loc.py function Dataset_Loc (line 5) | def Dataset_Loc(dataset, fname, partial=True, bn=False, few_shot=False): FILE: OcCo_Torch/utils/Inference_Timer.py class Inference_Timer (line 5) | class Inference_Timer: method __init__ (line 6) | def __init__(self, args): method update_args (line 24) | def update_args(self): method single_step (line 27) | def single_step(self, model, data): method update_single_epoch (line 38) | def update_single_epoch(self, logger): FILE: OcCo_Torch/utils/LMDB_DataFlow.py function resample_pcd (line 9) | def resample_pcd(pcd, n): class PreprocessData (line 17) | class PreprocessData(dataflow.ProxyDataFlow): method __init__ (line 19) | def __init__(self, ds, input_size, output_size): method get_data (line 24) | def get_data(self): class BatchData (line 31) | class BatchData(dataflow.ProxyDataFlow): method __init__ (line 32) | def __init__(self, ds, batch_size, input_size, gt_size, remainder=Fals... method __len__ (line 40) | def __len__(self): method __iter__ (line 49) | def __iter__(self): method _aggregate_batch (line 60) | def _aggregate_batch(self, data_holder, use_list=False): function lmdb_dataflow (line 75) | def lmdb_dataflow(lmdb_path, batch_size, input_size, output_size, is_tra... FILE: OcCo_Torch/utils/LMDB_Writer.py function sample_from_mesh (line 8) | def sample_from_mesh(filename, num_samples=16384): class pcd_df (line 13) | class pcd_df(DataFlow): method __init__ (line 14) | def __init__(self, model_list, num_scans, partial_dir, complete_dir, n... method size (line 21) | def size(self): method read_pcd (line 25) | def read_pcd(filename): method get_data (line 29) | def get_data(self): FILE: OcCo_Torch/utils/ModelNetDataLoader.py function pc_normalize (line 8) | def pc_normalize(pc): function farthest_point_sample (line 16) | def farthest_point_sample(point, npoint): class ModelNetDataLoader (line 40) | class ModelNetDataLoader(Dataset): method __init__ (line 41) | def __init__(self, root, npoint=1024, split='train', uniform=False, no... method __len__ (line 63) | def __len__(self): method _get_item (line 66) | def _get_item(self, index): method __getitem__ (line 89) | def __getitem__(self, index): class General_CLSDataLoader_HDF5 (line 93) | class General_CLSDataLoader_HDF5(Dataset): method __init__ (line 94) | def __init__(self, file_list, num_point=1024): method loadh5DataFile (line 112) | def loadh5DataFile(PathtoFile): method __len__ (line 116) | def __len__(self): method __getitem__ (line 119) | def __getitem__(self, index): class ModelNetJigsawDataLoader (line 125) | class ModelNetJigsawDataLoader(Dataset): method __init__ (line 126) | def __init__(self, root=r'./data/modelnet40_ply_hdf5_2048/jigsaw', method loadh5DataFile (line 156) | def loadh5DataFile(PathtoFile): method __getitem__ (line 160) | def __getitem__(self, index): method __len__ (line 165) | def __len__(self): FILE: OcCo_Torch/utils/PC_Augmentation.py function pc_normalize (line 12) | def pc_normalize(pc): function farthest_point_sample (line 21) | def farthest_point_sample(point, npoint): function random_shift_point_cloud (line 39) | def random_shift_point_cloud(batch_data, shift_range=0.1): function random_scale_point_cloud (line 48) | def random_scale_point_cloud(batch_data, scale_low=0.8, scale_high=1.25): function random_point_dropout (line 57) | def random_point_dropout(batch_pc, max_dropout_ratio=0.875): function translate_pointcloud_dgcnn (line 67) | def translate_pointcloud_dgcnn(pointcloud): function jitter_pointcloud_dgcnn (line 75) | def jitter_pointcloud_dgcnn(pointcloud, sigma=0.01, clip=0.02): FILE: OcCo_Torch/utils/S3DISDataLoader.py class S3DISDataset_HDF5 (line 20) | class S3DISDataset_HDF5(Dataset): method __init__ (line 23) | def __init__(self, root='data/indoor3d_sem_seg_hdf5_data', split='trai... method getDataFiles (line 56) | def getDataFiles(list_filename): method loadh5DataFile (line 60) | def loadh5DataFile(PathtoFile): method __getitem__ (line 64) | def __getitem__(self, index): method __len__ (line 70) | def __len__(self): class S3DISDataset (line 74) | class S3DISDataset(Dataset): method __init__ (line 76) | def __init__(self, root, block_points=4096, split='train', test_area=5... method __getitem__ (line 120) | def __getitem__(self, index): method __len__ (line 158) | def __len__(self): class S3DISDatasetWholeScene (line 162) | class S3DISDatasetWholeScene: method __init__ (line 163) | def __init__(self, root, block_points=8192, split='val', test_area=5, ... method __getitem__ (line 198) | def __getitem__(self, index): method __len__ (line 240) | def __len__(self): class ScannetDatasetWholeScene_evaluation (line 244) | class ScannetDatasetWholeScene_evaluation: method __init__ (line 246) | def __init__(self, root=root, block_points=8192, split='test', test_ar... method chunks (line 286) | def chunks(l, n): method split_data (line 292) | def split_data(data, idx): method nearest_dist (line 299) | def nearest_dist(block_center, block_center_list): method __getitem__ (line 306) | def __getitem__(self, index): method __len__ (line 401) | def __len__(self): FILE: OcCo_Torch/utils/ShapeNetDataLoader.py class PartNormalDataset (line 9) | class PartNormalDataset(Dataset): method __init__ (line 13) | def __init__(self, root, num_point=2048, split='train', use_normal=Fal... method read_fns (line 67) | def read_fns(path): method __getitem__ (line 72) | def __getitem__(self, index): method __len__ (line 90) | def __len__(self): FILE: OcCo_Torch/utils/TSNE_Visu.py function parse_args (line 15) | def parse_args(): FILE: OcCo_Torch/utils/Torch_Utility.py function seed_torch (line 6) | def seed_torch(seed=1029): function copy_parameters (line 17) | def copy_parameters(model, pretrained, verbose=True): function weights_init (line 36) | def weights_init(m): function bn_momentum_adjust (line 48) | def bn_momentum_adjust(m, momentum): FILE: OcCo_Torch/utils/TrainLogger.py class TrainLogger (line 7) | class TrainLogger: method __init__ (line 9) | def __init__(self, args, name='model', subfold='cls', filename='train_... method setup (line 24) | def setup(self, filename='train_log'): method mkdir (line 42) | def mkdir(self): method epoch_init (line 61) | def epoch_init(self, training=True): method step_update (line 66) | def step_update(self, pred, gt, loss, training=True): method epoch_update (line 75) | def epoch_update(self, training=True, mode='cls'): method epoch_summary (line 102) | def epoch_summary(self, writer=None, training=True, mode='cls'): method calculate_IoU (line 130) | def calculate_IoU(self): method train_summary (line 141) | def train_summary(self, mode='cls'): method update_from_checkpoints (line 152) | def update_from_checkpoints(self, checkpoint): FILE: OcCo_Torch/utils/Visu_Utility.py function plot_pcd_three_views (line 12) | def plot_pcd_three_views(filename, pcds, titles, suptitle='', sizes=None... FILE: OcCo_Torch/utils/gen_indoor3d_h5.py function insert_batch (line 39) | def insert_batch(data, label, last_batch=False): FILE: OcCo_Torch/utils/indoor3d_util.py function collect_point_label (line 36) | def collect_point_label(anno_path, out_filename, file_format='txt'): function data_to_obj (line 79) | def data_to_obj(data, name='example.obj', no_wall=True): function point_label_to_obj (line 90) | def point_label_to_obj(input_filename, out_filename, label_color=True, e... function sample_data (line 120) | def sample_data(data, num_sample): function sample_data_label (line 138) | def sample_data_label(data, label, num_sample): function room2blocks (line 145) | def room2blocks(data, label, num_point, block_size=1.0, stride=1.0, function room2blocks_plus (line 216) | def room2blocks_plus(data_label, num_point, block_size, stride, function room2blocks_wrapper (line 228) | def room2blocks_wrapper(data_label_filename, num_point, block_size=1.0, ... function room2blocks_plus_normalized (line 241) | def room2blocks_plus_normalized(data_label, num_point, block_size, stride, function room2blocks_wrapper_normalized (line 268) | def room2blocks_wrapper_normalized(data_label_filename, num_point, block... function room2samples (line 281) | def room2samples(data, label, sample_num_point): function room2samples_plus_normalized (line 318) | def room2samples_plus_normalized(data_label, num_point): function room2samples_wrapper_normalized (line 344) | def room2samples_wrapper_normalized(data_label_filename, num_point): function collect_bounding_box (line 359) | def collect_bounding_box(anno_path, out_filename): function bbox_label_to_obj (line 402) | def bbox_label_to_obj(input_filename, out_filename_prefix, easy_view=Fal... function bbox_label_to_obj_room (line 466) | def bbox_label_to_obj_room(input_filename, out_filename_prefix, easy_vie... function collect_point_bounding_box (line 546) | def collect_point_bounding_box(anno_path, out_filename, file_format): FILE: OcCo_Torch/utils/lmdb2hdf5.py function fix2len (line 8) | def fix2len(point_cloud, fix_length): FILE: render/Depth_Renderer.py function random_pose (line 8) | def random_pose(): function setup_blender (line 32) | def setup_blender(width, height, focal_length): FILE: render/EXR_Process.py function read_exr (line 11) | def read_exr(exr_path, height, width): function depth2pcd (line 21) | def depth2pcd(depth, intrinsics, pose): FILE: sample/mesh_sampling.cpp function uniform_deviate (line 53) | inline double function randomPointTriangle (line 60) | inline void function randPSurface (line 84) | inline void function uniform_sampling (line 112) | void function printHelp (line 159) | void function main (line 179) | int