Method__init__(self, in_dim, in_points, n_groups=1, query_dim=None, out_dim=None, out_points=None)
models/racformer_transformer.py:547
Method__init__(
self,
grid_config,
input_size,
downsample=16,
in_channels=512,
models/necks/view_transformer_racformer.py:37
Method__init__(self, inplanes, planes, kernel_size, padding, dilation,
norm_cfg=dict(type='BN'))
models/necks/view_transformer_racformer.py:331
Method__init__(self, in_ch, stage_ch, concat_ch, layer_per_block, module_name, SE=False, identity=False, depthwise=False, wi
models/backbones/vovnet.py:182
Method__init__(self, in_ch, stage_ch, concat_ch, block_per_stage, layer_per_block, stage_num, SE=False, depthwise=False, wit
models/backbones/vovnet.py:242
Method__init__(
self,
numC_input,
num_layer=[2, 2, 2],
num_channels=None,
models/backbones/resnet.py:19
Methodassign(self,
bbox_pred,
cls_pred,
gt_bboxes,
gt_labels,
models/bbox/assigners/polar_hungarian_assigner_3d.py:29
Functionbev_pool_v2_backwardFunction: pillar pooling (backward, cuda)
Args:
out_grad : grad of output bev feature, FloatTensor[b, c, h_out, w_out]
depth_grad
models/csrc/bev_pool_v2/src/bev_pool.cpp:74
Functionbev_pool_v2_forwardFunction: pillar pooling (forward, cuda)
Args:
depth : input depth, FloatTensor[n, d, h, w]
feat : input features, Fl
models/csrc/bev_pool_v2/src/bev_pool.cpp:30
Methodforward(self, query_bbox, query_feat, mlvl_feats, lss_bev_feats, radar_bev_feats, attn_mask, img_metas)
models/racformer_transformer.py:50
Methodforward(self, query_bbox, query_feat, mlvl_feats, lss_bev_feats, radar_bev_feats, attn_mask, img_metas)
models/racformer_transformer.py:91
Methodforward query_bbox: [B, Q, 10] [cx, cy, cz, w, h, d, rot.sin, rot.cos, vx, vy]
models/racformer_transformer.py:235
Methodforward(self, query_ray, query_feat, mlvl_feats, img_metas, d_region=0.1)
models/racformer_transformer.py:419
Methodforward(self, query_ray, query_feat, bev_feats, img_metas, d_region=0.1)
models/racformer_transformer.py:538
Methodforward(self, x, radar_depth, radar_rcs, img_metas, mlp_input)
models/necks/view_transformer_racformer.py:680