Method__init__(
self,
ann_file,
pipeline=None,
data_root=None,
classes=None,
menet/datasets/nuscenes_dataset.py:125
Method__init__(
self,
is_train,
final_dim=[256, 704],
resize_lim=(-0.06, 0.11),
bo
menet/datasets/pipelines/transforms_3d.py:94
Method__init__(self,
rot_range=[-0.78539816, 0.78539816],
scale_ratio_range=[0.95, 1.05],
menet/datasets/pipelines/transforms_3d.py:217
Method__init__(
self,
data_root: str,
xbound: Tuple[float, float, float],
ybound: Tuple[floa
menet/datasets/pipelines/loading.py:90
Method__init__(
self,
info_path,
data_root,
rate,
prepare,
sample_groups,
menet/datasets/pipelines/dpsampler.py:372
Method__init__(self,
numC_input,
numC_middle=None,
num_layer=[2,2,2],
menet/models/backbone/resnet.py:14
Method__init__(
self,
encoders: List[Dict],
fuser: Dict[str, Any],
decoder: Dict[str, Any],
menet/models/fusion_models/bevfusion.py:25
Functionbev_pool_backwardFunction: pillar pooling (backward, cuda)
Args:
out_grad : input features, FloatTensor[b, d, h, w, c]
geom_feats : input coord
menet/ops/bev_pool/src/bev_pool.cpp:60
Methodforward(ctx, x, geom_feats, ranks, B, D, H, W)
menet/ops/bev_pool/bev_pool.py:39