<reponame>mjseong0414/3d_detection_exercise
from .pv_rcnn import PVRCNN

__all__ = {
    'PVRCNN' : PVRCNN
}

def build_detector(model_cfg, num_class, dataset):
    model = __all__[model_cfg.NAME](
        model_cfg=model_cfg, num_class=num_class, dataset=dataset
    )

    return model