MCPcopy Create free account

hub / github.com/BIT-MJY/CVTNet / functions

Functions45 in github.com/BIT-MJY/CVTNet

↓ 6 callersFunctionload_files
(folder)
tools/utils/utils.py:167
↓ 5 callersMethod__init__
(self, d_model, eps=1e-6)
modules/cvtnet.py:18
↓ 4 callersFunctionrange_projection
(current_vertex, fov_up=10.67, fov_down=-30.67, proj_H=32, proj_W=900, max_range=80, cut_range=True,
tools/utils/utils.py:42
↓ 4 callersFunctionread_one_ri_bev_from_seq
(file_num, ri_bev_root)
tools/read_samples.py:18
↓ 3 callersMethodeval
(self)
test/test_cvtnet_prepare.py:37
↓ 2 callersFunctionbest_pos_distance
(query, pos_vecs)
modules/loss.py:4
↓ 2 callersFunctionbev_projection
(current_vertex, proj_H=32, proj_W=900, max_range=80, cut_height=True, lower_bound=10, upper_bound=20)
tools/utils/utils.py:105
↓ 2 callersFunctiongen_bev_image
CVTNet_libtorch/ws/fast_cvtnet.cpp:79
↓ 2 callersFunctiongen_range_image
CVTNet_libtorch/ws/fast_cvtnet.cpp:34
↓ 2 callersFunctionload_calib
(calib_path)
tools/utils/utils.py:24
↓ 2 callersFunctionload_poses
(pose_path)
tools/utils/utils.py:4
↓ 2 callersFunctionload_vertex
(scan_path)
tools/utils/utils.py:158
↓ 2 callersMethodtrain
(self)
train/train_cvtnet.py:48
↓ 1 callersMethod__init__
(self, feature_size, max_samples, cluster_size, output_dim, gating=True, add_batch_norm=True,
modules/netvlad.py:7
↓ 1 callersFunctionattention
(q, k, v, d_k, mask=None, dropout=None)
modules/cvtnet.py:32
↓ 1 callersFunctioncal_topn_recall
(recog_list_path, ground_truth_file_name, ri_bev_query_root, topn)
test/cal_topn_recall.py:16
↓ 1 callersFunctioncom_overlap_yaw
(scan_paths, poses, frame_idx, leg_output_width=360)
tools/utils/com_overlap_yaw.py:3
↓ 1 callersFunctiondata2xyzi
(data, flip=True)
tools/gen_ri_bev.py:23
↓ 1 callersMethodeval
(self)
test/test_yaw_rotation_invariance.py:36
↓ 1 callersMethodforward
(self, x)
modules/cvtnet.py:26
↓ 1 callersFunctionget_velo
(velofile)
tools/gen_ri_bev.py:35
↓ 1 callersFunctionread_lidar_data
CVTNet_libtorch/ws/fast_cvtnet.cpp:22
↓ 1 callersFunctionread_one_batch_ri_bev_from_seq
(f1_index, f1_seq, train_imgf1, train_imgf2, train_dir1, train_dir2, train_o
tools/read_samples.py:33
↓ 1 callersFunctionread_rotated_one_ri_bev_from_seq
(file_num, ri_bev_root, divd)
tools/read_samples.py:24
↓ 1 callersFunctiontriplet_loss
(q_vec, pos_vecs, neg_vecs, margin, use_min=False, lazy=False, ignore_zero_loss=False)
modules/loss.py:14
Method__init__
(self, heads, d_model, dropout=0.1)
modules/cvtnet.py:49
Method__init__
(self, d_model, d_ff=1024, dropout=0.1)
modules/cvtnet.py:83
Method__init__
(self, d_model, heads, dropout=0.1)
modules/cvtnet.py:98
Method__init__
(self, channels=5, use_transformer = True)
modules/cvtnet.py:117
Method__init__
(self, channels=5, use_transformer = True)
modules/cvtnet.py:176
Method__init__
(self, dim, add_batch_norm=True)
modules/netvlad.py:74
Method__init__
(self, channels=1, ri_bev_database_root=None, ri_bev_query_root=None, pretrained_weights=None, topn=50, save_r
test/test_cvtnet_prepare.py:22
Method__init__
(self, channels=1, ri_bev_database_root=None, ri_bev_query_root=None, pretrained_weights=None, topn=50, save_r
test/test_yaw_rotation_invariance.py:21
Method__init__
(self, channels=1, lr = 0.001, step_size=5, gamma=0.9, overlap_th=0.3, use_shuffle=False, num
train/train_cvtnet.py:22
Methodforward
(self, q, k, v, mask=None)
modules/cvtnet.py:63
Methodforward
(self, x)
modules/cvtnet.py:90
Methodforward
(self, x, mask)
modules/cvtnet.py:107
Methodforward
(self, x_l)
modules/cvtnet.py:145
Methodforward
(self, x_ri_bev)
modules/cvtnet.py:222
Methodforward
(self, x)
modules/netvlad.py:40
Methodforward
(self, x)
modules/netvlad.py:90
Functionmain
CVTNet_libtorch/ws/fast_cvtnet.cpp:127
Functionoverlap_orientation_npz_file2string_string_nparray
(npzfilenames, shuffle=True)
tools/read_all_sets.py:7
Functiontriplet_loss_inv
(q_vec, pos_vecs, neg_vecs, margin, use_min=True, lazy=False, ignore_zero_loss=False)
modules/loss.py:41
Functiontriplet_loss_wrapper
(q_vec, pos_vecs, neg_vecs, m1, m2, use_min=False, lazy=False, ignore_zero_loss=False)
modules/loss.py:68