14 lines
410 B
Python
14 lines
410 B
Python
|
import mmcv
|
||
|
|
|
||
|
|
from . import voxel_generator
|
||
|
|
|
||
|
|
|
||
|
|
def build_voxel_generator(cfg, **kwargs):
|
||
|
|
"""Builder of voxel generator."""
|
||
|
|
if isinstance(cfg, voxel_generator.VoxelGenerator):
|
||
|
|
return cfg
|
||
|
|
elif isinstance(cfg, dict):
|
||
|
|
return mmcv.runner.obj_from_dict(cfg, voxel_generator, default_args=kwargs)
|
||
|
|
else:
|
||
|
|
raise TypeError("Invalid type {} for building a sampler".format(type(cfg)))
|