MCPcopy Create free account

hub / github.com/cxmomo/RaCFormer / functions

Functions355 in github.com/cxmomo/RaCFormer

↓ 1 callersMethodinit_weights
Default initialization for Parameters of Module.
models/bev_self_attention.py:105
↓ 1 callersMethodinner_forward
query_bbox: [B, Q, 10] query_feat: [B, Q, C]
models/racformer_transformer.py:292
↓ 1 callersMethodinner_forward
query_bbox: [B, Q, 10] query_feat: [B, Q, C]
models/racformer_transformer.py:357
↓ 1 callersMethodinner_forward
query_bbox: [B, Q, 10] query_feat: [B, Q, C]
models/racformer_transformer.py:475
↓ 1 callersMethodinner_forward
(self, x, query)
models/racformer_transformer.py:576
↓ 1 callersMethodinner_forward
(self, bev_feats)
models/racformer_transformer.py:641
↓ 1 callersFunctioninverse_sigmoid
Inverse function of sigmoid. Args: x (Tensor): The tensor to do the inverse. eps (float): EPS avoid numerical
models/utils.py:86
↓ 1 callersFunctionlidar_render
(sample_token, data, out_path=None)
tools/visual.py:272
↓ 1 callersMethodload_offline
(self, results)
loaders/pipelines/loading.py:546
↓ 1 callersMethodload_offline
(self, results)
loaders/pipelines/loading.py:616
↓ 1 callersMethodload_offline
(self, results)
loaders/pipelines/loading.py:823
↓ 1 callersMethodloss
(self, gt_bboxes_list, gt_labels_list, preds_dicts, gt_bbo
models/racformer_head.py:430
↓ 1 callersMethodloss_single
(self, cls_scores, bbox_preds, gt_bboxes_list,
models/racformer_head.py:374
↓ 1 callersFunctionmain
()
val.py:50
↓ 1 callersFunctionmain
()
train.py:20
↓ 1 callersFunctionmake_cuda_ext
(name, module, sources, sources_cuda=[],
models/csrc/setup.py:6
↓ 1 callersFunctionmsmv_sampling
(mlvl_feats, sampling_locations, scale_weights)
models/csrc/wrapper.py:145
↓ 1 callersFunctionmsmv_sampling_pytorch
value: [B, N, H1W1 + H2W2..., C] sampling_locations: [B, Q, P, 3] scale_weights: [B, Q, P, 4]
models/csrc/wrapper.py:15
↓ 1 callersFunctionmsmv_sampling_pytorch_v2
value: [B, N, H1W1 + H2W2..., C] sampling_locations: [B, Q, P, 3] scale_weights: [B, Q, P, 4]
models/csrc/wrapper.py:41
↓ 1 callersFunctionmsmv_sampling_v2
(mlvl_feats, sampling_locations, scale_weights)
models/csrc/wrapper.py:155
↓ 1 callersFunctionone_hot
r"""Convert an integer label x-D tensor to a one-hot (x+1)-D tensor. Args: labels: tensor with labels of shape :math:`(N, *)`, where N is
models/necks/focalloss.py:12
↓ 1 callersMethodpoints2depthmap
(self, points, height, width)
loaders/pipelines/loading.py:323
↓ 1 callersMethodpoints2depthmap
(self, points, height, width)
loaders/pipelines/loading.py:471
↓ 1 callersMethodpre_compute
(self, img, img_metas)
models/necks/view_transformer_racformer.py:262
↓ 1 callersMethodprepare_for_dn_input
(self, batch_size, init_query_bbox, label_enc, img_metas)
models/racformer_head.py:136
↓ 1 callersMethodprepare_for_dn_loss
(self, mask_dict)
models/racformer_head.py:249
↓ 1 callersMethodradar_voxelize
Apply dynamic voxelization to points. Args: points (list[torch.Tensor]): Points of each sample. Returns: tup
models/racformer.py:151
↓ 1 callersMethodrefine_bbox
(self, bbox_proposal, bbox_delta)
models/racformer_transformer.py:226
↓ 1 callersFunctionrender_sample_data
Render sample data onto axis. :param sample_data_token: Sample_data token. :param with_anns: Whether to draw box annotations. :param
tools/visual.py:337
↓ 1 callersFunctionrgb_to_hsv
r"""Convert an image from RGB to HSV. .. image:: _static/img/rgb_to_hsv.png The image data is assumed to be in the range of (0, 1). Arg
models/utils.py:123
↓ 1 callersMethodrotate_z
(self, results, rot_angle)
loaders/pipelines/transforms.py:370
↓ 1 callersMethodrotate_z
(self, results, rot_angle)
loaders/pipelines/transforms.py:428
↓ 1 callersFunctionrotation_3d_in_axis
(points, angles)
models/utils.py:48
↓ 1 callersMethodsample_augmentation
https://github.com/Megvii-BaseDetection/BEVStereo/blob/master/dataset/nusc_mv_det_dataset.py#L247
loaders/pipelines/transforms.py:314
↓ 1 callersFunctionsampling_4d
Args: sample_points: 3D sampling points in shape [B, Q, T, G, P, 3] mlvl_feats: list of multi-scale features from neck, each in s
models/sparsebev_sampling.py:28
↓ 1 callersMethodsave_is_dict
(self, runner)
models/hook/is_save.py:68
↓ 1 callersMethodscale_xyz
(self, results, scale_ratio)
loaders/pipelines/transforms.py:385
↓ 1 callersMethodscale_xyz
(self, results, scale_ratio)
loaders/pipelines/transforms.py:449
↓ 1 callersMethodset_temporal_flag
(self, runner)
models/hook/sequentialsontrol.py:16
↓ 1 callersMethodsimple_test
(self, img_metas, img=None, rescale=False, radar_points=None, radar_depth=None, radar_rcs=None, **kwargs)
models/racformer.py:460
↓ 1 callersMethodsimple_test_offline
(self, img_metas, img=None, rescale=False, radar_points=None, radar_depth=None, radar_rcs=None)
models/racformer.py:463
↓ 1 callersMethodview_transform_core
(self, x, depth_digit, tran_feat, img_metas)
models/necks/view_transformer_racformer.py:268
↓ 1 callersFunctionvisualize_sample_radar
Visualizes a sample from BEV with annotations and detection results. :param nusc: NuScenes object. :param sample_token: The nuScenes samp
tools/render.py:109
↓ 1 callersMethodvoxel_pooling_v2
(self, coor, depth, feat)
models/necks/view_transformer_racformer.py:176
FunctionPYBIND11_MODULE
models/csrc/msmv_sampling copy/msmv_sampling.cpp:363
FunctionPYBIND11_MODULE
models/csrc/msmv_sampling/msmv_sampling.cpp:499
FunctionPYBIND11_MODULE
models/csrc/bev_pool_v2/src/bev_pool.cpp:106
Method__call__
Call function to pad images, masks, semantic segmentation maps. Args: results (dict): Result dict from loading pipeline. R
loaders/pipelines/transforms.py:51
Method__call__
Call function to normalize images. Args: results (dict): Result dict from loading pipeline. Returns: dict: Nor
loaders/pipelines/transforms.py:85
Method__call__
Call function to perform photometric distortion on images. Args: results (dict): Result dict from loading pipeline. Return
loaders/pipelines/transforms.py:147
Method__call__
(self, results)
loaders/pipelines/transforms.py:225
Method__call__
(self, results)
loaders/pipelines/transforms.py:355
Method__call__
(self, results)
loaders/pipelines/transforms.py:407
Method__call__
Call function to normalize images. Args: results (dict): Result dict from loading pipeline. Returns: dict: N
loaders/pipelines/transforms.py:483
Method__call__
Call function to flip bounding boxes, masks, semantic segmentation maps. Args: results (dict): Result dict from loading p
loaders/pipelines/transforms.py:600
Method__call__
(self, results)
loaders/pipelines/loading.py:63
Method__call__
Call function to load points data from file. Args: results (dict): Result dict containing point clouds data. Returns:
loaders/pipelines/loading.py:216
Method__call__
Call function to load points data from file. Args: results (dict): Result dict containing point clouds data. Returns:
loaders/pipelines/loading.py:371
Method__call__
(self, results)
loaders/pipelines/loading.py:492
Method__call__
(self, results)
loaders/pipelines/loading.py:593
Method__call__
(self, results)
loaders/pipelines/loading.py:721
Method__call__
Call function to load points data from file. Args: results (dict): Result dict containing point clouds data. Returns:
loaders/pipelines/loading.py:769
Method__call__
(self, results)
loaders/pipelines/loading.py:937
Method__call__
Call function to transform and format common fields in results. Args: results (dict): Result dict contains the data to convert.
loaders/pipelines/formatng.py:28
Method__call__
Call function to perform photometric distortion on images. Args: results (dict): Result dict from loading pipeline. Return
models/utils.py:248
Method__call__
Args: bbox_pred (Tensor): Predicted boxes with normalized coordinates (cx, cy, w, h), which are all in range [0,
models/bbox/match_costs/match_cost.py:15
Method__call__
Args: bbox_pred (Tensor): Predicted boxes with normalized coordinates (cx, cy, w, h), which are all in range [0,
models/bbox/match_costs/match_cost.py:39
Method__call__
(self, bboxes, gt_bboxes)
models/bbox/match_costs/match_cost.py:72
Method__call__
(self, iou)
models/bbox/match_costs/match_cost.py:87
Method__init__
(self, log_dir=None, interval=10, ignore_last=True, reset_flag=False, by_epoch=True)
utils.py:137
Method__init__
(self, data_root, info_file, ann_file, pip
loaders/vod_mono_dataset.py:35
Method__init__
(self, size=None, size_divisor=None, pad_val=0)
loaders/pipelines/transforms.py:22
Method__init__
(self, mean, std, to_rgb=True)
loaders/pipelines/transforms.py:80
Method__init__
(self, brightness_delta=32, contrast_range=(0.5, 1.5), satu
loaders/pipelines/transforms.py:137
Method__init__
(self, ida_aug_conf=None, training=True)
loaders/pipelines/transforms.py:221
Method__init__
(self, rot_range=[-0.3925, 0.3925], scale_ratio_range=[0.95, 1.05],
loaders/pipelines/transforms.py:347
Method__init__
(self, rot_range=[-0.3925, 0.3925], scale_ratio_range=[0.95, 1.05],
loaders/pipelines/transforms.py:399
Method__init__
(self, in_img_field=True)
loaders/pipelines/transforms.py:480
Method__init__
(self, flip_ratio=None, direction='horizontal')
loaders/pipelines/transforms.py:540
Method__init__
(self, prev_sweeps_num=5, next_sweeps_num=5, color_type='co
loaders/pipelines/loading.py:43
Method__init__
(self, coord_type, load_dim=6, use_dim=[0, 1, 2],
loaders/pipelines/loading.py:172
Method__init__
(self, grid_config, coord_type, load_dim=6,
loaders/pipelines/loading.py:297
Method__init__
(self, grid_config, downsample=1)
loaders/pipelines/loading.py:467
Method__init__
(self, grid_config, downsample=1, test_mode=False)
loaders/pipelines/loading.py:516
Method__init__
(self, sweeps_num=5, color_type='color', test_mode=False)
loaders/pipelines/loading.py:600
Method__init__
(self, coord_type, num_sweeps=5, norm_time=False,
loaders/pipelines/loading.py:751
Method__init__
(self, sweeps_num=5, filter=True, num_aggr_sweeps=6,
loaders/pipelines/loading.py:807
Method__init__
(self, class_names, with_gt=True, with_label=True)
loaders/pipelines/formatng.py:25
Method__init__
(self, data_aug=None, stop_prev_grad=False, pts_voxel_layer
models/racformer.py:20
Method__init__
(self, ratio=0.5, prob=0.7)
models/utils.py:9
Method__init__
(self, brightness_delta=32, contrast_range=(0.5, 1.5), satu
models/utils.py:238
Method__init__
(self)
models/utils.py:309
Method__init__
(self, *args, num_classes, in_channels, qu
models/racformer_head.py:15
Method__init__
(self, embed_dims=256, num_heads=8, num_levels=4,
models/bev_self_attention.py:51
Method__init__
(self, embed_dims, num_frames=8, num_points=4,
models/racformer_transformer.py:19
Method__init__
(self, embed_dims, num_frames=8, num_points=4,
models/racformer_transformer.py:60
Method__init__
(self, embed_dims, num_frames=8, num_points=4,
models/racformer_transformer.py:142
Method__init__
(self, embed_dims=256, num_heads=8, dropout=0.1, pc_range=[], init_cfg=None)
models/racformer_transformer.py:280
Method__init__
(self, embed_dims=256, num_frames=4, num_groups=4, num_points=8, num_levels=4, depth_num=15, pc_range=[], init
models/racformer_transformer.py:336
Method__init__
(self, embed_dims=256, num_frames=4, num_points=8,
models/racformer_transformer.py:427
← previousnext →101–200 of 355, ranked by callers