MCPcopy Create free account

hub / github.com/cattaneod/PointNetVlad-Pytorch / functions

Functions46 in github.com/cattaneod/PointNetVlad-Pytorch

⨍ Functions 46◇ Types & classes 6
↓ 11 callersFunctionload_pc_files
(filenames)
loading_pointclouds.py:35
↓ 10 callersFunctionlog_string
(out_str)
train_pointnetvlad.py:129
↓ 7 callersFunctionload_pc_file
(filename)
loading_pointclouds.py:23
↓ 5 callersMethod__init__
(self, num_points=2500, k=3, use_bn=True)
models/PointNetVlad.py:125
↓ 4 callersFunctionconstruct_query_and_database_sets
(base_path, runs_folder, folders, pointcloud_fols, filename, p, output_name)
generating_queries/generate_test_sets.py:51
↓ 4 callersFunctionjitter_point_cloud
Randomly jitter points. jittering is per point. Input: BxNx3 array, original batch of point clouds Return: BxNx3
loading_pointclouds.py:71
↓ 4 callersFunctionrotate_point_cloud
Randomly rotate the point clouds to augument the dataset rotation is per shape based along up direction Input: BxNx3 array,
loading_pointclouds.py:47
↓ 3 callersFunctionget_query_tuple
(dict_value, num_pos, num_neg, QUERY_DICT, hard_neg=[], other_neg=False)
loading_pointclouds.py:85
↓ 2 callersFunctionbest_pos_distance
(query, pos_vecs)
loss/pointnetvlad_loss.py:6
↓ 2 callersFunctioncheck_in_test_set
(northing, easting, points, x_width, y_width)
generating_queries/generate_test_sets.py:35
↓ 2 callersFunctioncheck_in_test_set
(northing, easting, points, x_width, y_width)
generating_queries/generate_training_tuples_refine.py:32
↓ 2 callersFunctionconstruct_query_dict
(df_centroids, filename)
generating_queries/generate_training_tuples_baseline.py:47
↓ 2 callersFunctionget_feature_representation
(filename, model)
train_pointnetvlad.py:345
↓ 2 callersFunctionget_latent_vectors
(model, dict_to_process)
evaluate.py:105
↓ 2 callersFunctionget_queries_dict
(filename)
loading_pointclouds.py:7
↓ 2 callersFunctionget_random_hard_negatives
(query_vec, random_negs, num_to_take)
train_pointnetvlad.py:364
↓ 2 callersFunctionget_sets_dict
(filename)
loading_pointclouds.py:15
↓ 2 callersFunctionoutput_to_file
(output, filename)
generating_queries/generate_test_sets.py:45
↓ 1 callersFunctioncheck_in_test_set
(northing, easting, points, x_width, y_width)
generating_queries/generate_training_tuples_baseline.py:37
↓ 1 callersFunctionconstruct_query_dict
(df_centroids, filename)
generating_queries/generate_training_tuples_refine.py:43
↓ 1 callersFunctionevaluate
()
evaluate.py:32
↓ 1 callersFunctionevaluate_model
(model)
evaluate.py:48
↓ 1 callersFunctionget_bn_decay
(batch)
train_pointnetvlad.py:122
↓ 1 callersFunctionget_latent_vectors
(model, dict_to_process)
train_pointnetvlad.py:379
↓ 1 callersFunctionget_learning_rate
(epoch)
train_pointnetvlad.py:137
↓ 1 callersFunctionget_recall
(m, n, DATABASE_VECTORS, QUERY_VECTORS, QUERY_SETS)
evaluate.py:165
↓ 1 callersFunctionrun_model
(model, queries, positives, negatives, other_neg, require_grad=True)
train_pointnetvlad.py:443
↓ 1 callersFunctiontrain
()
train_pointnetvlad.py:143
↓ 1 callersFunctiontrain_one_epoch
(model, optimizer, train_writer, loss_function, epoch)
train_pointnetvlad.py:208
↓ 1 callersFunctiontriplet_loss
(q_vec, pos_vecs, neg_vecs, margin, use_min=False, lazy=False, ignore_zero_loss=False)
loss/pointnetvlad_loss.py:15
Method__init__
(self, feature_size, max_samples, cluster_size, output_dim, gating=True, add_batch_norm=True,
models/PointNetVlad.py:13
Method__init__
(self, dim, add_batch_norm=True)
models/PointNetVlad.py:85
Method__init__
(self)
models/PointNetVlad.py:117
Method__init__
(self, num_points=2500, global_feat=True, feature_transform=False, max_pool=True)
models/PointNetVlad.py:181
Method__init__
(self, num_points=2500, global_feat=True, feature_transform=False, max_pool=True, output_dim=1024)
models/PointNetVlad.py:236
Functioncfg_str
()
config.py:50
Methodforward
(self, x)
models/PointNetVlad.py:45
Methodforward
(self, x)
models/PointNetVlad.py:101
Methodforward
(self, input)
models/PointNetVlad.py:120
Methodforward
(self, x)
models/PointNetVlad.py:150
Methodforward
(self, x)
models/PointNetVlad.py:201
Methodforward
(self, x)
models/PointNetVlad.py:244
Functionget_jittered_tuple
(dict_value, num_pos, num_neg, QUERY_DICT, hard_neg=[], other_neg=False)
loading_pointclouds.py:205
Functionget_rotated_tuple
(dict_value, num_pos, num_neg, QUERY_DICT, hard_neg=[], other_neg=False)
loading_pointclouds.py:145
Functionquadruplet_loss
(q_vec, pos_vecs, neg_vecs, other_neg, m1, m2, use_min=False, lazy=False, ignore_zero_loss=False)
loss/pointnetvlad_loss.py:49
Functiontriplet_loss_wrapper
(q_vec, pos_vecs, neg_vecs, other_neg, m1, m2, use_min=False, lazy=False, ignore_zero_loss=False)
loss/pointnetvlad_loss.py:45