| 1234567891011121314 |
- __all__ = ['build_head']
- def build_head(config):
- # det backbone
- from .db_head import DBHead
- support_dict = ['DBHead']
- module_name = config.pop('name')
- assert module_name in support_dict, Exception(
- 'det head only support {}'.format(support_dict))
- module_class = eval(module_name)(**config)
- return module_class
|
