MCPcopy Create free account

hub / github.com/DerrickXuNu/v2x-vit / functions

Functions300 in github.com/DerrickXuNu/v2x-vit

↓ 1 callersMethodload_img
()
v2xvit/models/sub_modules/torch_transformation_utils.py:370
↓ 1 callersMethodload_raw_transformation_matrix
(N)
v2xvit/models/sub_modules/torch_transformation_utils.py:377
↓ 1 callersMethodload_raw_transformation_matrix2
(N, alpha)
v2xvit/models/sub_modules/torch_transformation_utils.py:386
↓ 1 callersMethodlogging
Print out the loss function for current iteration. Parameters ---------- epoch : int Current epoch for
v2xvit/loss/pixor_loss.py:64
↓ 1 callersFunctionmain
()
v2xvit/tools/train.py:26
↓ 1 callersFunctionmain
()
v2xvit/tools/inference.py:37
↓ 1 callersFunctionmask_boxes_outside_range_numpy
Parameters ---------- boxes: np.ndarray (N, 7) [x, y, z, dx, dy, dz, heading], (x, y, z) is the box center limit_range: list
v2xvit/utils/box_utils.py:360
↓ 1 callersFunctionnormalize_homography
r""" Normalize a given homography in pixels to [-1, 1]. Args: dst_pix_trans_src_pix : torch.Tensor Homography/ies from sou
v2xvit/models/sub_modules/torch_transformation_utils.py:217
↓ 1 callersMethodnormalize_targets
Normalize label_map Parameters ---------- label_map : numpy.array Targets array for classification and r
v2xvit/data_utils/post_processor/bev_postprocessor.py:141
↓ 1 callersMethodpost_process_debug
Process the outputs of the model to 2D bounding box for debug purpose. Step1: convert each cav's output to bounding box format
v2xvit/data_utils/post_processor/bev_postprocessor.py:343
↓ 1 callersMethodproject_points_to_bev_map
Project points to BEV occupancy map with default ratio=0.1. Parameters ---------- points : np.ndarray (N
v2xvit/data_utils/datasets/basedataset.py:481
↓ 1 callersMethodreform_param
Reform the data params with current timestamp object groundtruth and delay timestamp LiDAR pose. Parameters --------
v2xvit/data_utils/datasets/basedataset.py:380
↓ 1 callersFunctionregroup
Regroup the data based on the record_len. Parameters ---------- dense_feature : torch.Tensor N, C, H, W record_len : lis
v2xvit/models/sub_modules/fuse_utils.py:8
↓ 1 callersMethodregroup
(self, x, record_len)
v2xvit/models/sub_modules/self_attn.py:54
↓ 1 callersMethodregroup
(self, x, record_len)
v2xvit/models/sub_modules/f_cooper_fuse.py:12
↓ 1 callersMethodregroup
(self, x, record_len)
v2xvit/models/sub_modules/v2v_fuse.py:43
↓ 1 callersMethodsigmoid_cross_entropy_with_logits
PyTorch Implementation for tf.nn.sigmoid_cross_entropy_with_logits: max(x, 0) - x * z + log(1 + exp(-abs(x))) in https://www.
v2xvit/loss/point_pillar_loss.py:176
↓ 1 callersFunctiontest
()
v2xvit/loss/pixor_loss.py:94
↓ 1 callersFunctiontest_bev_post_processing
()
v2xvit/tools/debug_utils.py:22
↓ 1 callersMethodtest_combine_roi_and_cav_mask
()
v2xvit/models/sub_modules/torch_transformation_utils.py:406
↓ 1 callersFunctiontest_parser
()
v2xvit/tools/debug_utils.py:12
↓ 1 callersFunctiontest_parser
()
v2xvit/tools/inference.py:16
↓ 1 callersMethodtime_delay_calculation
Calculate the time delay for a certain vehicle. Parameters ---------- ego_flag : boolean Whether the cur
v2xvit/data_utils/datasets/basedataset.py:325
↓ 1 callersFunctiontrain_parser
()
v2xvit/tools/train.py:15
↓ 1 callersMethodunpad_prior_encoding
(self, x, record_len)
v2xvit/models/point_pillar_v2vnet.py:70
↓ 1 callersMethodupdate_label_map
Update label_map based on bbx and regression targets. Parameters ---------- label_map : numpy.array Targ
v2xvit/data_utils/post_processor/bev_postprocessor.py:80
↓ 1 callersFunctionvis_parser
()
v2xvit/visualization/vis_data_sequence.py:11
↓ 1 callersMethodvisualize
Visualize the BEV 2D prediction, ground truth with point cloud together. Parameters ---------- pred_box_tensor : tor
v2xvit/data_utils/post_processor/bev_postprocessor.py:413
↓ 1 callersMethodvisualize_result
(self, pred_box_tensor, gt_tensor, pcd,
v2xvit/data_utils/datasets/basedataset.py:572
↓ 1 callersFunctionvisualize_single_sample_dataloader
Visualize a single frame of a single CAV for validation of data pipeline. Parameters ---------- o3d_pcd : o3d.PointCloud Ope
v2xvit/visualization/vis_utils.py:433
↓ 1 callersFunctionvoc_ap
VOC 2010 Average Precision.
v2xvit/utils/eval_utils.py:10
Method__getitem__
Abstract method, needs to be define by the children class.
v2xvit/data_utils/datasets/basedataset.py:175
Method__getitem__
(self, idx)
v2xvit/data_utils/datasets/intermediate_fusion_dataset.py:31
Method__getitem__
(self, idx)
v2xvit/data_utils/datasets/early_fusion_dataset.py:28
Method__getitem__
(self, idx)
v2xvit/data_utils/datasets/early_fusion_vis_dataset.py:25
Method__getitem__
(self, idx)
v2xvit/data_utils/datasets/late_fusion_dataset.py:30
Method__init__
(self, args)
v2xvit/models/point_pillar_opv2v.py:12
Method__init__
(self, args)
v2xvit/models/point_pillar_fcooper.py:12
Method__init__
(self, args)
v2xvit/models/point_pillar_transformer.py:14
Method__init__
(self, args)
v2xvit/models/point_pillar.py:13
Method__init__
(self, args)
v2xvit/models/point_pillar_v2vnet.py:13
Method__init__
(self, dim, heads, dim_heads, drop_out, window_size, relative_pos_embedding, fuse_method='nai
v2xvit/models/sub_modules/mswin.py:84
Method__init__
Initialize the ConvLSTM cell :param input_size: (int, int) Height and width of input tensor as (height, width). :
v2xvit/models/sub_modules/convgru.py:8
Method__init__
(self, dim, heads, num_types=2, num_relations=4, dim_head=64, dropout=0.1)
v2xvit/models/sub_modules/hmsa.py:8
Method__init__
(self, model_cfg, input_channels)
v2xvit/models/sub_modules/base_bev_backbone.py:7
Method__init__
(self, input_dim, compress_raito)
v2xvit/models/sub_modules/naive_compress.py:6
Method__init__
(self, dim)
v2xvit/models/sub_modules/self_attn.py:25
Method__init__
(self)
v2xvit/models/sub_modules/f_cooper_fuse.py:9
Method__init__
(self, config)
v2xvit/models/sub_modules/downsample_conv.py:33
Method__init__
(self, args)
v2xvit/models/sub_modules/v2v_fuse.py:15
Method__init__
(self, model_cfg, num_point_features, voxel_size, point_cloud_range)
v2xvit/models/sub_modules/pillar_vfe.py:57
Method__init__
(self, dim, hidden_dim, dropout=0.)
v2xvit/models/sub_modules/base_transformer.py:18
Method__init__
(self, dim, heads, dim_head=64, dropout=0.1)
v2xvit/models/sub_modules/base_transformer.py:36
Method__init__
(self, dim, depth, heads, dim_head, mlp_dim, dropout=0.)
v2xvit/models/sub_modules/base_transformer.py:84
Method__init__
(self, args)
v2xvit/models/sub_modules/base_transformer.py:104
Method__init__
(self)
v2xvit/models/sub_modules/torch_transformation_utils.py:366
Method__init__
(self, model_cfg)
v2xvit/models/sub_modules/point_pillar_scatter.py:6
Method__init__
(self, radix, cardinality)
v2xvit/models/sub_modules/split_attn.py:7
Method__init__
(self, args)
v2xvit/models/sub_modules/v2xvit_basic.py:12
Method__init__
(self, n_hid, RTE_ratio, max_len=100, dropout=0.2)
v2xvit/models/sub_modules/v2xvit_basic.py:40
Method__init__
(self, num_blocks, cav_att_config, pwindow_config)
v2xvit/models/sub_modules/v2xvit_basic.py:83
Method__init__
(self, args)
v2xvit/models/sub_modules/v2xvit_basic.py:124
Method__init__
(self, args)
v2xvit/models/sub_modules/v2xvit_basic.py:182
Method__init__
(self, anchor_params, train=True)
v2xvit/data_utils/post_processor/base_postprocessor.py:29
Method__init__
(self, anchor_params, train)
v2xvit/data_utils/post_processor/bev_postprocessor.py:17
Method__init__
(self, anchor_params, train)
v2xvit/data_utils/post_processor/voxel_postprocessor.py:19
Method__init__
(self, augment_config, train=True)
v2xvit/data_utils/augmentor/data_augmentor.py:24
Method__init__
(self, preprocess_params, train)
v2xvit/data_utils/pre_processor/base_preprocessor.py:19
Method__init__
(self, preprocess_params, train)
v2xvit/data_utils/pre_processor/sp_voxel_preprocessor.py:16
Method__init__
(self, preprocess_params, train)
v2xvit/data_utils/pre_processor/voxel_preprocessor.py:14
Method__init__
(self, preprocess_params, train)
v2xvit/data_utils/pre_processor/bev_preprocessor.py:11
Method__init__
(self, params, visualize, train=True)
v2xvit/data_utils/datasets/basedataset.py:44
Method__init__
(self, params, visualize, train=True)
v2xvit/data_utils/datasets/intermediate_fusion_dataset.py:21
Method__init__
(self, params, visualize, train=True)
v2xvit/data_utils/datasets/early_fusion_dataset.py:22
Method__init__
(self, params, visualize, train=True)
v2xvit/data_utils/datasets/early_fusion_vis_dataset.py:19
Method__init__
(self, params, visualize, train=True)
v2xvit/data_utils/datasets/late_fusion_dataset.py:24
Method__init__
(self, args)
v2xvit/loss/voxel_net_loss.py:7
Method__init__
(self, args)
v2xvit/loss/pixor_loss.py:9
Method__init__
Args: beta: Scalar float. L1 to L2 change point. For beta values < 1e-5, L1 loss is computed.
v2xvit/loss/point_pillar_loss.py:16
Method_check_kernel_size_consistency
(kernel_size)
v2xvit/models/sub_modules/convgru.py:186
Functionbbx2aabb
Convert the torch tensor bounding box to o3d aabb for visualization. Parameters ---------- bbx_center : torch.Tensor shape:
v2xvit/visualization/vis_utils.py:111
Functionbox3d_to_2d
Convert 3D bounding box to 2D. Parameters ---------- box3d : np.ndarray (n, 8, 3) Returns ------- box2d : np.nd
v2xvit/utils/box_utils.py:187
Functionboxes2d_to_corners2d
0 -------- 1 | | | | | | 3 -------- 2 Parameters __________ boxes2d: np.ndarray
v2xvit/utils/box_utils.py:101
Functionboxes_to_corners2d
0 -------- 1 | | | | | | 3 -------- 2 Parameters __________ boxes3d: np.ndarray
v2xvit/utils/box_utils.py:76
Functioncaluclate_tp_fp
Calculate the true positive and false positive numbers of the current frames. Parameters ---------- det_boxes : torch.Tensor
v2xvit/utils/eval_utils.py:36
Functioncheck_contain_nan
(x)
v2xvit/utils/common_utils.py:16
Methodcollate_batch
Customized collate function for target label generation. Parameters ---------- label_batch_list : list T
v2xvit/data_utils/post_processor/bev_postprocessor.py:190
Methodcollate_batch
Customized collate function for target label generation. Parameters ---------- label_batch_list : list T
v2xvit/data_utils/post_processor/voxel_postprocessor.py:196
Methodcollate_batch
Customized pytorch data loader collate function. Parameters ---------- batch : list or dict List or dict
v2xvit/data_utils/pre_processor/sp_voxel_preprocessor.py:59
Methodcollate_batch
Customized pytorch data loader collate function. Parameters ---------- batch : list or dict List or dict
v2xvit/data_utils/pre_processor/voxel_preprocessor.py:71
Methodcollate_batch_test
(self, batch)
v2xvit/data_utils/datasets/intermediate_fusion_dataset.py:375
Methodcollate_batch_test
Customized collate function for pytorch dataloader during testing for late fusion dataset. Parameters ----------
v2xvit/data_utils/datasets/early_fusion_dataset.py:183
Methodcollate_batch_test
Customized collate function for pytorch dataloader during testing for late fusion dataset. Parameters ----------
v2xvit/data_utils/datasets/late_fusion_dataset.py:161
Methodcollate_batch_train
Customized collate function for pytorch dataloader during training for late fusion dataset. Parameters ----------
v2xvit/data_utils/datasets/basedataset.py:517
Methodcollate_batch_train
Customized collate function for pytorch dataloader during training for late fusion dataset. Parameters ----------
v2xvit/data_utils/datasets/early_fusion_vis_dataset.py:151
Functioncompute_iou
Compute iou between box and boxes list Parameters ---------- box : shapely.geometry.Polygon Bounding box Polygon. boxes
v2xvit/utils/common_utils.py:119
Functionconvert_format
Convert boxes array to shapely.geometry.Polygon format. Parameters ---------- boxes_array : np.ndarray (N, 4, 2) or (N, 8, 3)
v2xvit/utils/common_utils.py:142
Functioncorner2d_to_standup_box
Find the minmaxx, minmaxy for each 2d box. (N, 4, 2) -> (N, 4) x1, y1, x2, y2 Parameters ---------- box2d : np.ndarray (
v2xvit/utils/box_utils.py:205
Functioncorner_to_standup_box_torch
Find the minmax x and y for each bounding box. Parameters ---------- box_corner : torch.Tensor Shape: (N, 8, 3) or (N, 4)
v2xvit/utils/box_utils.py:231
Functioncreate_loss
Create the loss function based on the given loss name. Parameters ---------- hypes : dict Configuration params for training.
v2xvit/tools/train_utils.py:118
← previousnext →101–200 of 300, ranked by callers