diff --git a/pcdet/models/backbones_3d/vfe/vfe_template.py b/pcdet/models/backbones_3d/vfe/vfe_template.py new file mode 100644 index 0000000..a862e3e --- /dev/null +++ b/pcdet/models/backbones_3d/vfe/vfe_template.py @@ -0,0 +1,22 @@ +import torch.nn as nn + + +class VFETemplate(nn.Module): + def __init__(self, model_cfg, **kwargs): + super().__init__() + self.model_cfg = model_cfg + + def get_output_feature_dim(self): + raise NotImplementedError + + def forward(self, **kwargs): + """ + Args: + **kwargs: + + Returns: + batch_dict: + ... + vfe_features: (num_voxels, C) + """ + raise NotImplementedError